Rejestrowanie nowych modeli
Rejestr Modeli MLflow umożliwia również rejestrowanie modeli podczas przebiegu trenowania. Jest to wygodne, ponieważ pozwala na zalogowanie i zarejestrowanie modelu za pomocą tej samej funkcji.
W tym ćwiczeniu użyjesz smaku scikit-learn, aby zarejestrować model w Rejestrze Modeli podczas przebiegu trenowania – w momencie logowania modelu do MLflow Tracking. Następnie przeszukasz Rejestr Modeli, aby upewnić się, że model został pomyślnie zarejestrowany.
Model zostanie zarejestrowany obok już istniejących modeli wytrenowanych na danych treningowych "Insurance". Model jest już wytrenowany i przypisany do zmiennej lr. Podczas przeszukiwania Rejestru Modeli instancja MLflowClient() jest dostępna pod nazwą client, a ciąg filtrujący został już utworzony jako insurance_filter_string.
To ćwiczenie jest częścią kursu
Wprowadzenie do MLflow
Instrukcje do ćwiczenia
- Zarejestruj model
lrpod nazwą"Insurance", używając smaku scikit-learn. - Przeszukaj Rejestr Modeli za pomocą
client, aby upewnić się, że model został zarejestrowany.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Log the model using scikit-learn flavor
____.____.____(____, "model", ____="____")
insurance_filter_string = "name = 'Insurance'"
# Search for Insurance models
print(____.____(____=____))