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.
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
- 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.
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.
# Create Python Class
class ____(mlflow.pyfunc.PythonModel):
# Set method for loading model
def ____(self, context):
self.model = mlflow.sklearn.load_model("./lr_model/")
# Set method for custom inference
def ____(self, context, model_input):
predictions = self.model.predict(model_input)
decoded_predictions = []
for prediction in predictions:
if prediction == 0:
decoded_predictions.append("female")
else:
decoded_predictions.append("male")
return decoded_predictions