MulaiMulai sekarang secara gratis

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

Lihat Kursus

Petunjuk latihan

  • Impor Flavor sklearn dari modul mlflow.
  • 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]]))
Edit dan Jalankan Kode