ComenzarEmpieza gratis

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

Ver curso

Ejercicio interactivo práctico

Pon en práctica la teoría con uno de nuestros ejercicios interactivos

Empezar ejercicio