1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Giới thiệu về MLflow
Connected

Bài tập

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.

Hướng dẫn

100 XP
  • 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.