CommencerCommencer gratuitement

Enregistrer de nouveaux modèles

Le MLflow Model Registry peut également enregistrer des modèles pendant une exécution d’entraînement. C’est utile, car cela permet de consigner et d’enregistrer un modèle avec une seule fonction.

Dans cet exercice, vous allez utiliser le flavor scikit-learn pour enregistrer un modèle dans le Model Registry pendant une exécution d’entraînement, lorsque le modèle est consigné dans MLflow Tracking. Vous rechercherez ensuite dans le Model Registry afin de vérifier que le modèle a bien été enregistré.

Ce modèle sera enregistré aux côtés d’autres modèles déjà enregistrés et entraînés sur les données d’entraînement "Insurance". Le modèle a déjà été entraîné et est stocké dans la variable lr. Pour la recherche dans le Model Registry, une instance de MLflowClient() a été affectée à client et la chaîne de filtre a déjà été créée sous le nom insurance_filter_string.

Cet exercice fait partie du cours

Introduction à MLflow

Afficher le cours

Instructions

  • Enregistrez le modèle lr sous "Insurance" en utilisant le flavor scikit-learn.
  • Interrogez le Model Registry avec client pour vérifier que le modèle a été enregistré.

Exercice interactif pratique

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

# Log the model using scikit-learn flavor
____.____.____(____, "model", ____="____")
insurance_filter_string = "name = 'Insurance'"

# Search for Insurance models
print(____.____(____=____))
Modifier et exécuter le code