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

Tải mô hình từ Model Registry

Trong bài tập này, bạn sẽ dùng flavor scikit-learn để triển khai mô hình "Insurance" ổn định nhất từ MLflow Model Registry và sau đó dùng dữ liệu kiểm thử để lấy dự đoán từ mô hình.

Mô hình sử dụng LogisticRegression để dự đoán một yêu cầu bồi thường bảo hiểm thuộc nam hay nữ, được gán nhãn 1 hoặc 0. Bạn sẽ tải mô hình và tạo dự đoán bằng tập kiểm thử X_test.

Mô-đun MLflow đã được nhập sẵn.

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

  • Với flavor scikit-learn, hãy tải phiên bản "Production" của mô hình "Insurance" bằng cách sử dụng quy ước lấy mô hình từ Registry làm model URI.
  • Dùng mô hình đã tải để chạy dự đoán trên dữ liệu kiểm thử từ train_test_split được dùng trong quá trình huấn luyện mô hình.

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.

# Load the Production stage of Insurance model using scikit-learn flavor
model = ____.____.____("____")

# Run prediction on our test data
____.____(____)
Chỉnh sửa và Chạy Mã