MulaiMulai sekarang secara gratis

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

Lihat Kursus

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
Edit dan Jalankan Kode