Añadir parámetros a MLproject
Definir parámetros en los Proyectos MLflow te permite hacer reproducible tu código ML. Los parámetros también simplifican la ejecución de experimentos de entrenamiento con diferentes ajustes sin tener que cambiar el código.
En este ejercicio, vas a añadir parámetros a tu archivo MLproject
para el punto de entrada principal. Este punto de entrada se utiliza para ejecutar el script train_model.py
que entrena un modelo de Regresión Logística a partir de los datos del Seguro.
El script acepta dos parámetros, n_jobs
y fit_intercept
, que son hiperparámetros utilizados para entrenar el modelo. Empezarás añadiendo el parámetro n_jobs
en el archivo MLproject
. A continuación, añadirás el parámetro fit_intercept
. Por último, añadirás los parámetros al comando ejecutado en el punto de entrada principal.
Este ejercicio forma parte del curso
Introducción a MLflow
Instrucciones de ejercicio
- Crea un parámetro llamado
n_jobs
de tipoint
y valor por defecto1
. - Crea un segundo parámetro llamado
fit_intercept
como tipobool
con un valor por defecto fijado enTrue
. - Introduce ambos parámetros en el comando asegurándote de que
n_jobs
es el primero seguido defit_intercept
.
Ejercicio interactivo práctico
Pruebe este ejercicio completando este código de muestra.
"""
name: insurance_model
python_env: python_env.yaml
entry_points:
main:
parameters:
# Create parameter for number of jobs as n_jobs
____:
____: ____
____: ____
# Create parameter for fit_intercept
____:
____: ____
____: ____
# Add parameters to be passed into the command
command: "python3.9 train_model.py {____} {____}"
"""