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

Question

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

Answer 1

如果您在机器学习工作室（经典）上工作，模型可以在 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')

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