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

Question

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

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

def my_square(x):
    return x**2

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

Answer 1

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

在此容器中，您可以做任何您想做的事情，包括 return 您的 my_square 功能。请记住，您必须编写自己的 Flask 微服务，包括代理和 wsgi 服务器（如果需要）。

我认为 this example 是最有帮助的。