Zacznij terazZacznij za darmo

Zapisywanie i wczytywanie modelu

Dzięki Model API modele można udostępniać między deweloperami, którzy mogą nie mieć dostępu do tego samego serwera MLflow Tracking – wystarczy skorzystać z lokalnego systemu plików.

W tym ćwiczeniu wytrenujesz nowy model LinearRegression na podstawie istniejącego, używając zbioru danych Unicorn. Najpierw wczytasz istniejący model z lokalnego systemu plików. Następnie wytrenujesz nowy model na jego podstawie i zapiszesz go z powrotem w lokalnym systemie plików.

Istniejący model został zapisany w lokalnym systemie plików w katalogu o nazwie "lr_local_v1". Moduł mlflow zostanie zaimportowany.

To ćwiczenie jest częścią kursu

Wprowadzenie do MLflow

Zobacz kurs

Instrukcje do ćwiczenia

  • Wczytaj model z katalogu "lr_local_v1" w lokalnym systemie plików, używając biblioteki scikit-learn z modułu MLflow.
  • Korzystając z biblioteki scikit-learn z modułu mlflow, zapisz model lokalnie w katalogu o nazwie "lr_local_v2".

Interaktywne ćwiczenie praktyczne

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

# Load model from local filesystem
model = ____.____.____("____")

# Training Data
X = df[["R&D Spend", "Administration", "Marketing Spend", "State"]]
y = df[["Profit"]]
X_train, X_test, y_train, y_test = train_test_split(X, y, train_size=0.7,random_state=0)
# Train Model
model.fit(X_train, y_train)

# Save model to local filesystem
____.____.____(____, "____")
Edytuj i uruchom kod