如何解决如何在sagemaker中运行python脚本其中编写了一些功能?
我有python script.py
,它是从头开始的KNN机器学习算法编写者。
仅使用pandas和numpy库。
脚本包含所有功能。
它需要一些参数,例如train_data
,test_data
,k_val
等。我想在 sagemaker 中运行它。
寻找一种解决方案,如何将该脚本放入docker映像中,并从 ECR容器在 sagemaker 中调用它。
解决方法
您需要的是 SageMaker BYOC(自带容器)。
SageMaker为大多数ML算法预先构建了容器,但是对于自定义算法和自定义代码,它提供了创建自定义容器的支持。
它们基本上有2种类型:
如果您的目标是训练模型,请运行验证并保存模型文件。
-Training Container
如果您的目标是使用现有模型进行推断,请使用SageMaker Endpoint Hosting或Batch Transform。
-Serving Container
有关详细步骤,请参阅此repository。
文件predictor.py
和serve
最重要,其中包含您的自定义代码。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。