1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Úvod do MLflow
Connected

cvičení

Vytvoření vlastní Python třídy

MLflow umožňuje vytvářet vlastní modely, které pokryjí širokou škálu různých případů použití. K tomu slouží Python třída, která dědí třídu mlflow.pyfunc.PythonModel. Třída PythonModel nabízí flexibilitu díky metodám pro vlastní logiku inference a správu závislostí artefaktů.

V tomto cvičení vytvoříš novou Python třídu pro vlastní model, který načte konkrétní model a po inferenci dekóduje popisky. Modul mlflow bude již naimportován.

Pokyny

100 XP
  • Vytvoř Python třídu s názvem CustomPredict.
  • Definuj metodu load_context(), která slouží k načítání artefaktů ve vlastní třídě.
  • Definuj metodu predict() pro implementaci vlastní logiky inference.