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

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_testy_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

Xem khóa học

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

  • Log mô hình lr_class bằng flavor "built-in" của scikit-learn.
  • Gọi hàm evaluate() từ module mlflow.
  • Đánh giá tập dữ liệu eval_data và đặ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"
)
Chỉnh sửa và Chạy Mã