ÎncepețiÎncepe gratuit

Configurează antrenarea modelului folosind CML

În acest exercițiu, vei folosi GitHub Action CML pentru a antrena un clasificator Random Forest care să prezică precipitațiile. CML este un GitHub Action care simplifică generarea rapoartelor pentru experimente de ML.

Antrenarea va fi declanșată când deschizi un PR față de ramura main. Vei continua să lucrezi cu setul de date meteo; fișierul preprocess_dataset.py conține funcții ajutătoare pentru preprocesarea setului de date, la fel ca înainte.

Rezultatul rulării train.py este un fișier metrics.json cu metricile modelului și un fișier confusion_matrix.png cu graficul matricei de confuzie.

Sarcina ta este să completezi fișierul .github/workflows/train_cml.yaml pregătit parțial, pentru a defini un flux de antrenare a modelului la nivel înalt.

NOTĂ: Folosește python3 în loc de python pentru a rula scripturile Python.

Acest exercițiu face parte din cursul

CI/CD pentru Machine Learning

Vezi cursul

Exercițiu interactiv practic

Transformă teoria în practică cu unul dintre exercițiile noastre interactive

Începe exercițiul