使用 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。
本练习是课程的一部分
