amazon-web-services - 将一个简单的函数部署到 amazon sagemaker

Question

我一直在玩 Amazon Sagemaker。他们在不同领域都有很棒的样本笔记本。但是，出于测试目的，我想创建一个从函数返回结果的端点。据我目前所见，我的理解是我们只能部署模型，但我想澄清一下。

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

def my_square(x):
    return x**2

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

Answer 1

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

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

在我看来this example是最有帮助的。