Configurer l'entraînement du modèle avec CML
Dans cet exercice, vous utiliserez la GitHub Action CML pour entraîner un classifieur Random Forest afin de prédire les précipitations. CML est une GitHub Action qui simplifie la génération de rapports pour les expériences en ML.
L'entraînement se déclenchera lorsque vous ouvrirez une PR vers la branche main.
Vous continuerez à travailler avec l'ensemble de données météo; le fichier preprocess_dataset.py contient des fonctions utilitaires pour prétraiter l'ensemble de données comme précédemment.
L'exécution de train.py produit un fichier metrics.json contenant les mesures du modèle et un fichier confusion_matrix.png avec le graphique de la matrice de confusion.
Votre tâche consiste à compléter l'échafaudage de .github/workflows/train_cml.yaml afin de définir un flux d'entraînement de modèle de haut niveau.
REMARQUE : Utilisez python3 au lieu de python pour exécuter les scripts Python.
Cette activité fait partie du cours
CI/CD pour le Machine Learning
Exercice interactif pratique
Passez de la théorie à l’action grâce à l’un de nos exercices interactifs
Commencer l’exercice