1. Nauka
    2. /
  2. Kursy
    3. /
  3. Wprowadzenie do MLflow
Connected

ćwiczenie

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.

Instrukcje

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