Lägga till parametrar i MLproject
Att definiera parametrar i MLflow Projects gör din maskininlärningskod reproducerbar. Parametrar förenklar också körningen av träningsexperiment med olika inställningar – utan att du behöver ändra koden.
I den här övningen ska du lägga till parametrar i din MLproject-fil för huvud-entry-pointen. Den här entry pointen används för att köra skriptet train_model.py, som tränar en logistisk regressionsmodell på försäkringsdata.
Skriptet tar emot två parametrar, n_jobs och fit_intercept, som är hyperparametrar för modellträningen. Du börjar med att lägga till parametern n_jobs i MLproject-filen, lägger sedan till parametern fit_intercept och avslutar med att skicka båda parametrarna till kommandot som körs i huvud-entry-pointen.
Den här övningen är en del av kursen
Introduktion till MLflow
Övningsinstruktioner
- Skapa en parameter med namnet
n_jobsav typenintoch standardvärdet1. - Skapa en andra parameter med namnet
fit_interceptav typenboolmed standardvärdetTrue. - Skicka båda parametrarna till kommandot så att
n_jobskommer först, följt avfit_intercept.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
"""
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 {____} {____}"
"""