Este proyecto demuestra cómo automatizar el entrenamiento de un modelo de clustering KMeans en el conjunto de datos Iris y registrar los detalles del experimento utilizando Weights & Biases (WandB), todo esto integrado con GitHub Actions para una ejecución automática.
Asegúrate de tener configurado tu entorno de GitHub y WandB con las siguientes consideraciones:
- Una cuenta de GitHub.
- Una cuenta en WandB y una API KEY generada.
- Configurar la API KEY de WandB como un secreto en GitHub (
WANDB_API_KEY
).
El flujo de trabajo definido en .github/workflows
se activa con cada push al archivo source/load_n_classify.py
en la rama master
. El flujo incluye los siguientes pasos:
- Checkout del Código: Utiliza
actions/checkout@v3
para obtener el código más reciente. - Configuración de Python: Establece el entorno Python usando
actions/setup-python@v4
, específicamente la versión3.9
. - Instalación de Dependencias: Instala todas las dependencias necesarias definidas en
requirements.txt
. - Login en WandB: Realiza el login en WandB utilizando la API KEY almacenada en los secretos de GitHub.
- Ejecución del Script: Ejecuta
load_n_classify.py
pasando como argumento--IdExecution
el número de ejecución (run_number
) de GitHub Actions.
Para utilizar este flujo de trabajo:
- Realiza cambios en el script
source/load_n_classify.py
según sea necesario. - Haz push de tus cambios al archivo en la rama
master
. - GitHub Actions automáticamente ejecutará el flujo de trabajo, entrenando el modelo de clustering y registrando el experimento en WandB.
- El script registra los resultados del entrenamiento y las visualizaciones en tu proyecto de WandB.
- Puedes monitorear la ejecución y el resultado del flujo de trabajo en la pestaña
Actions
de tu repositorio GitHub.