Aan de slagBegin gratis

Verpak een Machine Learning-model

In deze oefening train je een LinearRegression-model uit scikit-learn om de winst van een Unicorn Company te voorspellen.

Je gebruikt de ingebouwde scikit-learn Flavor van MLflow om het model te verpakken. Je gebruikt de auto-loggingfunctie van de Flavor om automatisch metrieken, parameters en het model naar MLflow Tracking te loggen wanneer de estimator wordt gefit.

Deze oefening maakt deel uit van de cursus

Introductie tot MLflow

Bekijk cursus

Oefeninstructies

  • Importeer de sklearn-flavor uit de mlflow-module.
  • Stel het experiment in op "Sklearn Model".
  • Gebruik auto-logging van de flavor om je model te verpakken.

Interactieve oefening met praktijkervaring

Probeer deze oefening door deze voorbeeldcode aan te vullen.

# 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]]))
Code bewerken en uitvoeren