Membuat Kelas Python kustom
MLflow menyediakan cara untuk membuat model kustom agar dapat mendukung beragam kasus penggunaan. Untuk membuat model kustom, MLflow memungkinkan pengguna membuat sebuah Kelas Python yang mewarisi Kelas mlflow.pyfunc.PythonModel. Kelas PythonModel menyediakan kustomisasi dengan menyediakan metode untuk logika inferensi kustom dan dependensi artefak.
Pada latihan ini, Anda akan membuat Kelas Python baru untuk model kustom yang memuat model tertentu dan kemudian mendekode label setelah inferensi. Modul mlflow akan diimpor.
Latihan ini adalah bagian dari kursus
Pengantar MLflow
Petunjuk latihan
- Buat Kelas Python dengan nama
CustomPredict. - Definisikan metode
load_context()yang digunakan untuk memuat artefak dalam Kelas kustom. - Definisikan metode
predict()untuk mendefinisikan inferensi kustom.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# 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