Smak scikit-learn i ewaluacja modelu
W tym ćwiczeniu wytrenujemy model klasyfikacyjny i ocenimy jego działanie. Model korzysta ze zbioru danych Insurance Charges, aby sklasyfikować, czy dane dotyczą kobiety, czy mężczyzny.
Zaczniemy od zalogowania modelu do MLflow Tracking przy użyciu smaku scikit-learn, a następnie ocenimy model na zbiorze eval_data.
Zbiór ewaluacyjny jest dostępny jako eval_data, a wytrenowany model nosi nazwę lr_class. Zbiór eval_data składa się z X_test i y_test, ponieważ dane treningowe zostały podzielone za pomocą funkcji train_test_split() z biblioteki sklearn.
# Model
lr_class = LogisticRegression()
lr_class.fit(X_train, y_train)
Moduł mlflow jest już zaimportowany.
To ćwiczenie jest częścią kursu
Wprowadzenie do MLflow
Instrukcje do ćwiczenia
- Zaloguj model
lr_classprzy użyciu wbudowanego smaku scikit-learn. - Wywołaj funkcję
evaluate()z modułumlflow. - Oceń zbiór danych
eval_data, wskazując kolumnę"sex"jako wartości docelowe.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Eval Data
eval_data = X_test
eval_data["sex"] = y_test
# Log the lr_class model using Scikit-Learn Flavor
___.___.___(____, "model")
# Get run id
run = mlflow.last_active_run()
run_id = run.info.run_id
# Evaluate the logged model with eval_data data
___.___(f"runs:/{run_id}/model",
____=____,
____="____",
model_type="classifier"
)