Flavor scikit-learn và đánh giá
Trong bài tập này, bạn sẽ huấn luyện một mô hình phân loại và đánh giá hiệu năng của nó. Mô hình sử dụng tập dữ liệu Insurance Charges của bạn để phân loại chi phí thuộc về nữ hay nam.
Chúng ta sẽ bắt đầu bằng cách log mô hình lên MLflow Tracking bằng flavor scikit-learn và kết thúc bằng việc đánh giá mô hình của bạn sử dụng tập dữ liệu eval_data.
Tập dữ liệu đánh giá của bạn đã được tạo là eval_data và mô hình được huấn luyện có tên lr_class. eval_data bao gồm X_test và y_test vì dữ liệu huấn luyện đã được tách bằng hàm train_test_split() từ sklearn.
# Model
lr_class = LogisticRegression()
lr_class.fit(X_train, y_train)
Module mlflow đã được import.
Bài tập này là một phần của khóa học
Giới thiệu về MLflow
Hướng dẫn bài tập
- Log mô hình
lr_classbằng flavor "built-in" của scikit-learn. - Gọi hàm
evaluate()từ modulemlflow. - Đánh giá tập dữ liệu
eval_datavà đặt target là cột"sex".
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.
# Eval Data
eval_data = X_test
eval_data["sex"] = y_test
# Log the lr_class model using Scikit-Learn Flavor
___.___.___(____, "model")
# Get run id
run = mlflow.last_active_run()
run_id = run.info.run_id
# Evaluate the logged model with eval_data data
___.___(f"runs:/{run_id}/model",
____=____,
____="____",
model_type="classifier"
)