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
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
____.____(____)