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 merupakan bagian dari kursus
Pengantar MLflow
Instruksi latihan
- Impor Flavor
sklearndari modulmlflow. - Tetapkan Experiment ke
"Sklearn Model". - Gunakan auto logging dari flavor untuk mengemas model Anda.
Latihan interaktif langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
# 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]]))