Pacchetto di un modello di apprendimento automatico
In questo esercizio, addestrerai un modello di LinearRegression da scikit-learn per prevedere i profitti di un'azienda Unicorn.
Utilizzerai il Flavor scikit-learn integrato in MLflow per confezionare il modello. Utilizzerai la funzione di auto logging di Flavor per registrare automaticamente le metriche, i parametri e il modello in MLflow Tracking quando viene richiamato lo stimatore di fit.
Questo esercizio fa parte del corso
Introduzione a MLflow
Istruzioni dell'esercizio
- Importa il Flavor
sklearndal modulomlflow. - Imposta l'esperimento su
"Sklearn Model". - Utilizza la registrazione automatica del sapore per confezionare il tuo modello.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# Import Scikit-learn flavor
import mlflow.____
# Set the experiment to "Sklearn Model"
mlflow.____("____")
# Set Auto logging for Scikit-learn flavor
____.____.____()
lr = LinearRegression()
lr.fit(X_train, y_train)
# Get a prediction from test data
print(lr.predict(X_test.iloc[[5]]))