Bắt đầu ngayBắt đầu miễn phí

Đóng gói một mô hình Machine Learning

Trong bài tập này, bạn sẽ huấn luyện mô hình LinearRegression từ scikit-learn để dự đoán lợi nhuận của một công ty Unicorn.

Bạn sẽ dùng Flavor scikit-learn tích hợp sẵn của MLflow để đóng gói mô hình. Bạn cũng sẽ dùng hàm auto logging của Flavor để tự động ghi log các chỉ số (metrics), tham số (parameters) và mô hình lên MLflow Tracking khi bộ ước lượng (estimator) được fit.

Bài tập này là một phần của khóa học

Giới thiệu về MLflow

Xem khóa học

Hướng dẫn bài tập

  • Import Flavor sklearn từ module mlflow.
  • Đặt Experiment thành "Sklearn Model".
  • Sử dụng auto logging từ flavor để đóng gói mô hình của bạn.

Bài tập tương tác thực hành trực tiếp

Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.

# 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]]))
Chỉnh sửa và Chạy Mã