MulaiMulai sekarang secara gratis

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

Lihat Kursus

Petunjuk latihan

  • Catat (log) model lr_class menggunakan flavor bawaan scikit-learn.
  • Panggil fungsi evaluate() dari modul mlflow.
  • Evaluasi himpunan data eval_data dan 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"
)
Edit dan Jalankan Kode