作为数据科学团队的一员,您希望在实验阶段尝试不同的建模方法。为保证可再现性,每种方法都有您需要手动跟踪的不同参数。Agent Platform SDK for Python 自动日志记录是一种使用 MLflow 的单行代码 SDK 功能,它提供与 Gemini Enterprise Agent Platform 上的实验及实验运行作业关联的自动指标和参数跟踪。
笔记本:Gemini Enterprise Agent Platform 上的实验自动日志记录
在“Gemini Enterprise Agent Platform 上的实验:自动日志记录”笔记本中,您将了解如何使用 Gemini Enterprise Agent Platform 上的实验来执行以下操作:
- 在 Agent Platform SDK for Python 中启用自动日志记录功能。
- 训练 scikit-learn 模型,并查看在不设置实验运行作业的情况下,产生的实验运行作业的指标和参数是否自动记录到 Gemini Enterprise Agent Platform 上的实验。
- 训练 TensorFlow 模型,并通过使用
aiplatform.start_run()和aiplatform.end_run()手动设置实验运行作业,检查自动记录到 Gemini Enterprise Agent Platform 上的实验的指标和参数。 - 在 Agent Platform SDK for Python 中停用自动日志记录功能,训练一个 PyTorch 模型并检查是否没有记录任何参数或指标。