Začněte nyníZačněte zdarma

Ukládání a načítání modelu

S pomocí Model API lze modely sdílet mezi vývojáři, kteří nemusí mít přístup ke stejnému MLflow Tracking serveru – stačí použít lokální souborový systém.

V tomto cvičení natrénuješ nový model LinearRegression na základě již existujícího modelu za použití datasetu Unicorn. Nejprve načteš existující model z lokálního souborového systému. Poté z něj natrénuješ nový model a uložíš ho zpět na lokální souborový systém.

Existující model je uložen na lokálním souborovém systému v adresáři "lr_local_v1". Modul mlflow bude již naimportován.

Toto cvičení je součástí kurzu

Úvod do MLflow

Zobrazit kurz

Pokyny k cvičení

  • Načti model z adresáře "lr_local_v1" na lokálním souborovém systému pomocí scikit-learn knihovny z modulu MLflow.
  • Pomocí scikit-learn knihovny z modulu mlflow ulož model lokálně do adresáře "lr_local_v2".

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

# 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
____.____.____(____, "____")
Upravit a spustit kód