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

Bài tập

Tạo một Python Class tùy chỉnh

MLflow cung cấp cách tạo các mô hình tùy chỉnh để hỗ trợ nhiều trường hợp sử dụng khác nhau. Để tạo mô hình tùy chỉnh, MLflow cho phép bạn tạo một Python Class kế thừa Class mlflow.pyfunc.PythonModel. Class PythonModel cho phép tùy biến bằng cách cung cấp các phương thức cho logic suy luận tùy chỉnh và khai báo phụ thuộc vào artifact.

Trong bài tập này, bạn sẽ tạo một Python Class mới cho mô hình tùy chỉnh: mô hình này tải một mô hình cụ thể và giải mã nhãn sau khi suy luận. Module mlflow sẽ được import.

Hướng dẫn

100 XP
  • Tạo một Python Class với tên CustomPredict.
  • Định nghĩa phương thức load_context() dùng để tải artifacts trong Class tùy chỉnh.
  • Định nghĩa phương thức predict() để thiết lập logic suy luận tùy chỉnh.