使用 CML 設定模型訓練
在這個練習中,你會使用 CML GitHub Action 來訓練一個 Random Forest Classifier,用於預測降雨。CML 是一個 GitHub Action,可以簡化產生機器學習實驗報告的流程。
當你對 main 分支開啟 PR 時,訓練會被觸發。
你會持續使用 weather 資料集;preprocess_dataset.py 檔案包含用來如前處理資料集的輔助函式。
執行 train.py 的輸出包含一個 metrics.json 檔案(內含模型評估指標),以及一個 confusion_matrix.png 檔案(內含混淆矩陣的圖)。
你的任務是完成樣板檔 .github/workflows/train_cml.yaml,以定義高階的模型訓練流程。
注意:執行 Python 指令時請使用 python3,不要使用 python。
本練習屬於課程
