Cargar modelos desde el Registro de Modelos
En este ejercicio, utilizarás el sabor scikit-learn para desplegar el modelo "Insurance"
más estable del Registro de Modelos MLflow y, a continuación, utilizarás datos de prueba para obtener una predicción del modelo.
El modelo utiliza LogisticRegression para predecir si una reclamación al seguro corresponde a un hombre o a una mujer, que se etiqueta como 1 ó 0. Cargarás el modelo y luego harás predicciones utilizando un conjunto de pruebas llamado X_test
.
Se importará el módulo MLflow.
Este ejercicio forma parte del curso
Introducción a MLflow
Instrucciones de ejercicio
- Con el sabor scikit-learn, carga la versión
"Production"
del modelo"Insurance"
utilizando la convención para obtener modelos del Registro como el modelo URI. - Utilizando el modelo cargado, realiza una predicción sobre los datos de prueba de
train_test_split
utilizados durante el entrenamiento del modelo.
Ejercicio interactivo práctico
Pruebe este ejercicio completando este código de muestra.
# Load the Production stage of Insurance model using scikit-learn flavor
model = ____.____.____("____")
# Run prediction on our test data
____.____(____)