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
Oefeninstructies
- Importeer de
sklearn-flavor uit demlflow-module. - Stel het experiment in op
"Sklearn Model". - Gebruik auto-logging van de flavor om je model te verpakken.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in 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]]))