ÎncepețiÎncepe gratuit

Adăugarea parametrilor în MLproject

Definirea parametrilor în MLflow Projects îți permite să faci codul ML reproductibil. Parametrii simplifică și rularea experimentelor de antrenament cu setări diferite, fără a fi nevoie să modifici codul.

În acest exercițiu, vei adăuga parametri în fișierul MLproject pentru punctul de intrare principal. Acest punct de intrare este folosit pentru a rula scriptul train_model.py, care antrenează un model de Regresie Logistică pe date de tip Insurance.

Scriptul acceptă doi parametri, n_jobs și fit_intercept, care sunt hiperparametri utilizați pentru antrenarea modelului. Vei începe prin a adăuga parametrul n_jobs în fișierul MLproject, după care vei adăuga parametrul fit_intercept. În final, vei adăuga ambii parametri comenzii executate în punctul de intrare principal.

Acest exercițiu face parte din cursul

Introducere în MLflow

Vezi cursul

Instrucțiuni pentru exercițiu

  • Creează un parametru numit n_jobs de tip int cu valoarea implicită 1.
  • Creează un al doilea parametru numit fit_intercept de tip bool cu valoarea implicită True.
  • Transmite ambii parametri în command, asigurându-te că n_jobs este primul, urmat de fit_intercept.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

"""
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 {____} {____}"
"""
Editează și rulează codul