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
Petunjuk latihan
- Catat model ke MLflow Tracking di bawah jalur artefak
"lr_tracking". - Buat variabel bernama
runyang diatur ke run terakhir. - Buat variabel lain bernama
run_idyang diatur kerun_iddari variabelrun. - Muat model menggunakan
run_iddan 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:/{____}/____")