微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何保存或提取在 Azure ML Studio 中开发的机器学习模型?

如何解决如何保存或提取在 Azure ML Studio 中开发的机器学习模型?

所以我在 Azure ML Studio 中构建了我的机器学习预测模型。我想在我的 Web 应用程序中使用该模型,我将使用 Flask 和 Heroku 进行开发。

解决方法

如果你在机器学习工作室(经典)工作,模型可以在 AzureML 工作室中训练、评分、保存和运行,但不能下载到你的本地机器。无法对 AzureML 之外的模型执行任何操作。

作为这个链接:https://docs.microsoft.com/en-us/azure/machine-learning/how-to-train-pytorch

“您还可以使用 Run 对象下载模型的本地副本。在训练脚本 pytorch_train.py 中,PyTorch 保存对象将模型保存到本地文件夹(计算目标本地)。您可以使用下载副本的 Run 对象。”

# Create a model folder in the current directory
os.makedirs('./model',exist_ok=True)

# Download the model from run history
run.download_file(name='outputs/model.pt',output_file_path='./model/model.pt')

这也适用于不同的框架,请查看文档以获取更多详细信息和选项。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。