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
Instrukcje do ćwiczenia
- Zaimportuj flavor
sklearnz modułumlflow. - 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]]))