MulaiMulai sekarang secara gratis

Mencatat dan memuat model

Model API menyediakan cara untuk berinteraksi dengan model dengan mencatat (logging) dan memuatnya langsung dari MLflow Tracking secara terstandar. Kemampuan berinteraksi dengan model sangat penting selama siklus ML pada langkah Model Engineering dan Model Evaluation.

Pada latihan ini Anda akan membuat model Linear Regression dari scikit-learn menggunakan himpunan data Unicorn. Model ini akan dicatat ke MLflow Tracking dan kemudian dimuat menggunakan run_id yang digunakan untuk mencatat artefak.

Pertama, Anda akan mencatat model menggunakan pustaka scikit-learn dari modul MLflow. Lalu Anda akan memuat model dari MLflow Tracking menggunakan run_id.

Model akan dilatih dan diberi nama lr_model.

lr_model = LinearRegression()
lr_model.fit(X_train, y_train)

Modul mlflow akan diimpor.

Latihan ini adalah bagian dari kursus

Pengantar MLflow

Lihat Kursus

Petunjuk latihan

  • Catat model ke MLflow Tracking di bawah jalur artefak "lr_tracking".
  • Buat variabel bernama run yang diatur ke run terakhir.
  • Buat variabel lain bernama run_id yang diatur ke run_id dari variabel run.
  • Muat model menggunakan run_id dan jalur artefak yang digunakan untuk mencatat model.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Log model to MLflow Tracking
____.____.____(____, "____")

# Get the last run
run = ____.____

# Get the run_id of the above run
run_id = ____.___.____

# Load model from MLflow Tracking
model = ____.____.____(f"runs:/{____}/____")
Edit dan Jalankan Kode