Začněte nyníZačněte zdarma

Zabalení modelu strojového učení

V tomto cvičení natrénuješ model LinearRegression ze scikit-learn, který bude předpovídat zisk společnosti Unicorn Company.

Pro zabalení modelu využiješ vestavěný scikit-learn Flavor z MLflow. Pomocí funkce automatického logování daného Flavoru se metriky, parametry i model automaticky zalogují do MLflow Tracking při zavolání fit estimátoru.

Toto cvičení je součástí kurzu

Úvod do MLflow

Zobrazit kurz

Pokyny k cvičení

  • Importuj sklearn Flavor z modulu mlflow.
  • Nastav experiment na "Sklearn Model".
  • Použij automatické logování z daného flavoru pro zabalení modelu.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

# Import Scikit-learn flavor
import mlflow.____

# Set the experiment to "Sklearn Model"
mlflow.____("____")

# Set Auto logging for Scikit-learn flavor 
____.____.____()

lr = LinearRegression()
lr.fit(X_train, y_train)

# Get a prediction from test data
print(lr.predict(X_test.iloc[[5]]))
Upravit a spustit kód