首页 > 解决方案 > 将一个简单的功能部署到 amazon sagemaker

问题描述

我一直在玩 Amazon Sagemaker。他们在不同的领域都有惊人的样本笔记本。但是,出于测试目的,我想创建一个从函数返回结果的端点。从我目前看到的情况来看,我的理解是我们只能部署模型,但我想澄清一下。

假设我想调用端点,它应该给我输入值的平方。所以,我将首先创建一个函数:

def my_square(x):
    return x**2

我们可以在 Amazon Sagemaker 中部署这个简单的功能吗?

标签: amazon-web-servicesamazon-sagemaker

解决方案


是的,这是可能的,但需要一些开销:您可以将自己的 docker 图像传递给 sagemaker 进行训练和推理。

在这个容器中,你可以做任何你想做的事情,包括返回你的my_square函数。请记住,您必须编写自己的烧瓶微服务,包括代理和 wsgi 服务器(如果需要)。

在我看来,这个例子是最有帮助的。


推荐阅读