Zacznij terazZacznij za darmo

Dodawanie parametrów do MLproject

Definiowanie parametrów w projektach MLflow pozwala uczynić kod uczenia maszynowego reprodukowalnym. Parametry upraszczają też uruchamianie eksperymentów treningowych z różnymi ustawieniami – bez konieczności modyfikowania kodu.

W tym ćwiczeniu dodasz parametry do pliku MLproject dla głównego punktu wejścia. Ten punkt wejścia służy do uruchamiania skryptu train_model.py, który trenuje model regresji logistycznej na danych ubezpieczeniowych.

Skrypt przyjmuje dwa parametry: n_jobs oraz fit_intercept – są to hiperparametry używane podczas trenowania modelu. Najpierw dodasz parametr n_jobs do pliku MLproject, następnie parametr fit_intercept, a na końcu przekażesz oba parametry do polecenia wykonywanego w głównym punkcie wejścia.

To ćwiczenie jest częścią kursu

Wprowadzenie do MLflow

Zobacz kurs

Instrukcje do ćwiczenia

  • Utwórz parametr o nazwie n_jobs typu int z domyślną wartością 1.
  • Utwórz drugi parametr o nazwie fit_intercept typu bool z domyślną wartością True.
  • Przekaż oba parametry do polecenia, dbając o to, aby n_jobs był pierwszy, a po nim fit_intercept.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

"""
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 {____} {____}"
"""
Edytuj i uruchom kod