教程:Azure Databricks 上的端到端深度学习模型

本教程笔记本演示了在 Azure Databricks 中训练深度学习模型的端到端示例,包括加载数据、可视化数据、设置并行超参数优化,以及使用 MLflow 查看结果、注册模型,以及使用 Spark UDF 中的已注册模型对新数据执行推理。

该笔记本使用 PyTorch,这是一个 Python 包,它提供 GPU 加速张量计算和高级功能来构建深度学习网络。

准备就绪后,可以使用马赛克 AI 模型服务部署模型。

MLflow PyTorch 模型训练笔记本

获取笔记本