Configura el entrenamiento del modelo con CML
En este ejercicio, usarás CML GitHub Action para entrenar un Random Forest Classifier que prediga precipitaciones. CML es una GitHub Action que simplifica la generación de informes para experimentos de ML.
El entrenamiento se ejecutará cuando abras un PR contra la rama main.
Seguirás trabajando con el conjunto de datos meteorológicos; el archivo preprocess_dataset.py incluye funciones auxiliares para preprocesar el conjunto de datos como antes.
La ejecución de train.py genera un archivo metrics.json con las métricas del modelo y un archivo confusion_matrix.png con la gráfica de la matriz de confusión.
Tu tarea es completar el esqueleto de .github/workflows/train_cml.yaml para definir un flujo de entrenamiento de modelo de alto nivel.
NOTA: usa python3 en lugar de python para ejecutar los scripts de Python.
Este ejercicio forma parte del curso
CI/CD para Machine Learning
Ejercicio interactivo práctico
Pon en práctica la teoría con uno de nuestros ejercicios interactivos
Empezar ejercicio