CommencerCommencer gratuitement

Chargement de modèles à partir du registre de modèles

Dans cet exercice, vous utiliserez la saveur scikit-learn pour déployer le modèle "Insurance" le plus stable du MLflow Model Registry, puis vous utiliserez des données de test pour obtenir une prédiction à partir du modèle.

Le modèle utilise la régression logistique pour prédire si une demande d'assurance concerne un homme ou une femme, ce qui est étiqueté comme 1 ou 0. Vous chargerez le modèle et ferez ensuite des prédictions à l'aide d'un ensemble de tests appelé X_test.

Le module MLflow sera importé.

Cet exercice fait partie du cours

Introduction à MLflow

Afficher le cours

Instructions

  • Avec la saveur scikit-learn, chargez la version "Production" du modèle "Insurance" en utilisant la convention de récupération des modèles du Registre en tant que modèle URI.
  • À l'aide du modèle chargé, effectuez une prédiction sur les données de test de train_test_split utilisées lors de la formation du modèle.

Exercice interactif pratique

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

# Load the Production stage of Insurance model using scikit-learn flavor
model = ____.____.____("____")

# Run prediction on our test data
____.____(____)
Modifier et exécuter le code