CommencerCommencer gratuitement

Enregistrer et charger un modèle

Avec la Model API, les modèles peuvent être partagés entre développeurs qui n’ont pas accès au même serveur MLflow Tracking en utilisant un système de fichiers local.

Dans cet exercice, vous allez entraîner un nouveau modèle LinearRegression à partir d’un modèle existant en utilisant le jeu de données Unicorn. Vous commencerez par charger un modèle existant depuis le système de fichiers local. Ensuite, vous entraînerez un nouveau modèle à partir de ce modèle existant et vous l’enregistrerez de nouveau sur le système de fichiers local.

Le modèle existant a été enregistré sur le système de fichiers local dans un répertoire appelé "lr_local_v1". Le module mlflow sera importé.

Cet exercice fait partie du cours

Introduction à MLflow

Afficher le cours

Instructions

  • Chargez le modèle depuis le répertoire du système de fichiers local "lr_local_v1" en utilisant la bibliothèque scikit-learn du module MLflow.
  • En utilisant la bibliothèque scikit-learn du module mlflow, enregistrez le modèle localement dans un répertoire appelé "lr_local_v2".

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# 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
____.____.____(____, "____")
Modifier et exécuter le code