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@v3para 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.pypasando como argumento--IdExecutionel 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.pysegú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
Actionsde tu repositorio GitHub.