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

Question

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

Answer 1

如果您正在使用机器学习工作室(经典版)，则可以在 AzureML 工作室中对模型进行训练、评分、保存和运行，但无法将其下载到您的本地计算机。无法对 AzureML 之外的模型执行任何操作。

作为此链接:https://learn.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')

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