ÎncepețiÎncepe gratuit

Împachetează un model de învățare automată

În acest exercițiu, vei antrena un model LinearRegression din scikit-learn pentru a prezice profitul unei companii Unicorn.

Vei folosi Flavorul scikit-learn integrat în MLflow pentru a împacheta modelul. Vei utiliza funcția de auto logging a Flavorului pentru a înregistra automat metrici, parametri și modelul în MLflow Tracking atunci când este apelat estimatorul fit.

Acest exercițiu face parte din cursul

Introducere în MLflow

Vezi cursul

Instrucțiuni pentru exercițiu

  • Importă Flavorul sklearn din modulul mlflow.
  • Setează experimentul la "Sklearn Model".
  • Folosește funcția de auto logging din flavor pentru a împacheta modelul.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# 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]]))
Editează și rulează codul