Zacznij terazZacznij za darmo

Pakowanie modelu uczenia maszynowego

W tym ćwiczeniu wytrenujesz model LinearRegression z biblioteki scikit-learn, który będzie przewidywał zysk fikcyjnej firmy Unicorn Company.

Do spakowania modelu użyjesz wbudowanego w MLflow flavoru scikit-learn. Funkcja automatycznego rejestrowania tego flavoru pozwoli ci automatycznie zapisywać metryki, parametry i model w MLflow Tracking w momencie wywołania dopasowanego estymatora.

To ćwiczenie jest częścią kursu

Wprowadzenie do MLflow

Zobacz kurs

Instrukcje do ćwiczenia

  • Zaimportuj flavor sklearn z modułu mlflow.
  • Ustaw eksperyment na "Sklearn Model".
  • Użyj automatycznego rejestrowania z flavoru, aby spakować model.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

# 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]]))
Edytuj i uruchom kod