Flavor scikit-learn dan evaluasi
Dalam latihan ini Anda akan melatih sebuah model klasifikasi dan mengevaluasi kinerjanya. Model menggunakan himpunan data Insurance Charges untuk mengklasifikasikan apakah biaya tersebut untuk perempuan atau laki-laki.
Kita akan mulai dengan mencatat model ke MLflow Tracking menggunakan flavor scikit-learn dan diakhiri dengan mengevaluasi model Anda menggunakan himpunan data eval_data.
Himpunan data evaluasi Anda dibuat sebagai eval_data dan model kita dilatih dengan nama lr_class. eval_data terdiri dari X_test dan y_test karena data pelatihan dibagi menggunakan fungsi train_test_split() dari sklearn.
# Model
lr_class = LogisticRegression()
lr_class.fit(X_train, y_train)
Modul mlflow sudah diimpor.
Latihan ini adalah bagian dari kursus
Pengantar MLflow
Petunjuk latihan
- Catat (log) model
lr_classmenggunakan flavor bawaan scikit-learn. - Panggil fungsi
evaluate()dari modulmlflow. - Evaluasi himpunan data
eval_datadan targetkan kolom"sex".
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# 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"
)