Kom igångKom igång gratis

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

Visa kurs

Övningsinstruktioner

  • Skapa en parameter med namnet n_jobs av typen int och standardvärdet 1.
  • Skapa en andra parameter med namnet fit_intercept av typen bool med standardvärdet True.
  • Skicka båda parametrarna till kommandot så att n_jobs kommer först, följt av fit_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 {____} {____}"
"""
Redigera och kör kod