1. Nauka
  2. /
  3. Kursy
  4. /
  5. Wprowadzenie do MLflow

Connected

ćwiczenie

Tworzenie niestandardowej klasy Pythona

MLflow umożliwia tworzenie niestandardowych modeli, dzięki czemu można obsługiwać szeroką gamę przypadków użycia. W tym celu MLflow pozwala na definiowanie klas Pythona dziedziczących po klasie mlflow.pyfunc.PythonModel. Klasa PythonModel oferuje elastyczność poprzez metody służące do niestandardowej logiki wnioskowania oraz obsługi zależności od artefaktów.

W tym ćwiczeniu stworzysz nową klasę Pythona dla niestandardowego modelu, który wczytuje określony model, a następnie dekoduje etykiety po wykonaniu wnioskowania. Moduł mlflow zostanie zaimportowany.

Instrukcje

100 XP
  • Utwórz klasę Pythona o nazwie CustomPredict.
  • Zdefiniuj metodę load_context() służącą do ładowania artefaktów w niestandardowej klasie.
  • Zdefiniuj metodę predict() służącą do definiowania niestandardowej logiki wnioskowania.