Rejestrowanie i ładowanie modelu
Model API umożliwia interakcję z modelami poprzez ich rejestrowanie i ładowanie bezpośrednio z MLflow Tracking w ustandaryzowany sposób. Możliwość pracy z modelami jest kluczowa w cyklu życia uczenia maszynowego – zarówno na etapie budowania modeli, jak i ich ewaluacji.
W tym ćwiczeniu stworzysz model regresji liniowej z biblioteki scikit-learn, korzystając ze zbioru danych Unicorn. Model zostanie zarejestrowany w MLflow Tracking, a następnie załadowany przy użyciu run_id przypisanego do zarejestrowanego artefaktu.
Na początku zarejestrujesz model za pomocą biblioteki scikit-learn dostępnej w module MLflow. Następnie załadujesz model z MLflow Tracking przy użyciu run_id.
Model zostanie wytrenowany pod nazwą lr_model.
lr_model = LinearRegression()
lr_model.fit(X_train, y_train)
Moduł mlflow będzie już zaimportowany.
To ćwiczenie jest częścią kursu
Wprowadzenie do MLflow
Instrukcje do ćwiczenia
- Zarejestruj model w MLflow Tracking, używając ścieżki artefaktu
"lr_tracking". - Utwórz zmienną o nazwie
run, która będzie wskazywać na ostatnie uruchomienie. - Utwórz kolejną zmienną o nazwie
run_id, która będzie ustawiona na wartośćrun_idzmiennejrun. - Załaduj model, używając
run_idoraz ścieżki artefaktu użytej podczas rejestrowania modelu.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Log model to MLflow Tracking
____.____.____(____, "____")
# Get the last run
run = ____.____
# Get the run_id of the above run
run_id = ____.___.____
# Load model from MLflow Tracking
model = ____.____.____(f"runs:/{____}/____")