问题描述
所以我在 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')
这也适用于不同的框架,请查看文档以获取更多详细信息和选项。