开始使用免费开始使用

使用 CML 设置模型训练

在本练习中,您将使用 CML GitHub Action 训练一个随机森林分类器来预测降雨。CML 是一种 GitHub Action,用于简化生成机器学习实验的报告。

当您针对 main 分支打开 PR 时,将触发训练。 您将继续使用天气数据集;preprocess_dataset.py 文件包含与之前相同的数据预处理辅助函数。

运行 train.py 的输出包括一个包含模型指标的 metrics.json 文件,以及一个包含混淆矩阵可视化的 confusion_matrix.png 文件。

您的任务是完成脚手架文件 .github/workflows/train_cml.yaml,以制定高层次的模型训练流程。

注意:运行 Python 脚本时,请使用 python3,不要使用 python

本练习是课程的一部分

面向机器学习的 CI/CD

查看课程

动手互动练习

通过我们的互动练习之一,将理论转化为实践

开始练习