Thiết lập huấn luyện mô hình bằng CML
Trong bài tập này, bạn sẽ dùng CML GitHub Action để huấn luyện một Random Forest Classifier dự đoán lượng mưa. CML là một GitHub Action giúp đơn giản hóa việc tạo báo cáo cho các thử nghiệm ML.
Quá trình huấn luyện sẽ được kích hoạt khi bạn mở một PR vào nhánh main.
Bạn sẽ tiếp tục làm việc với bộ dữ liệu thời tiết; tệp preprocess_dataset.py chứa các hàm hỗ trợ tiền xử lý bộ dữ liệu như trước.
Đầu ra khi chạy train.py là một tệp metrics.json chứa các chỉ số của mô hình và tệp confusion_matrix.png chứa biểu đồ ma trận nhầm lẫn.
Nhiệm vụ của bạn là hoàn thiện khung .github/workflows/train_cml.yaml để xây dựng luồng huấn luyện mô hình ở mức cao.
LƯU Ý: Dùng python3 thay vì python để chạy các script Python.
Bài tập này là một phần của khóa học
CI/CD cho Machine Learning
Bài tập tương tác thực hành
Biến lý thuyết thành hành động với một trong các bài tập tương tác của chúng tôi
Bắt đầu bài tập