Kemaskan model machine learning
Pada latihan ini, Anda akan melatih model LinearRegression dari scikit-learn untuk memprediksi profit sebuah Perusahaan Unicorn.
Anda akan menggunakan Flavor scikit-learn bawaan MLflow untuk mengemas model. Anda akan menggunakan fungsi auto logging dari Flavor untuk mencatat metrik, parameter, dan model secara otomatis ke MLflow Tracking saat estimator fit dipanggil.
Latihan ini adalah bagian dari kursus
Pengantar MLflow
Petunjuk latihan
- Impor Flavor
sklearndari modulmlflow. - Tetapkan Experiment ke
"Sklearn Model". - Gunakan auto logging dari flavor untuk mengemas model Anda.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Import Scikit-learn flavor
import mlflow.____
# Set the experiment to "Sklearn Model"
mlflow.____("____")
# Set Auto logging for Scikit-learn flavor
____.____.____()
lr = LinearRegression()
lr.fit(X_train, y_train)
# Get a prediction from test data
print(lr.predict(X_test.iloc[[5]]))