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
Instrucțiuni pentru exercițiu
- Creează un parametru numit
n_jobsde tipintcu valoarea implicită1. - Creează un al doilea parametru numit
fit_interceptde tipboolcu valoarea implicităTrue. - Transmite ambii parametri în command, asigurându-te că
n_jobseste primul, urmat defit_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 {____} {____}"
"""