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
Instrukcje do ćwiczenia
- Utwórz parametr o nazwie
n_jobstypuintz domyślną wartością1. - Utwórz drugi parametr o nazwie
fit_intercepttypuboolz domyślną wartościąTrue. - Przekaż oba parametry do polecenia, dbając o to, aby
n_jobsbył pierwszy, a po nimfit_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 {____} {____}"
"""