IniziaInizia gratis

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

Visualizza il corso

Istruzioni dell'esercizio

  • Importa il Flavor sklearn dal modulo mlflow.
  • 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]]))
Modifica ed esegui il codice