1. Learn
    2. /
  2. Courses
    3. /
  3. Introduction to MLflow

Exercise

Package a machine learning model

In this exercise, you will train a LinearRegression model from scikit-learn to predict profit of a Unicorn Company.

You will use MLflow's built-in scikit-learn Flavor to package the model. You will use the Flavor's auto logging function to automatically log metrics, parameters and the model to MLflow Tracking when the fit estimator is called.

Instructions

100 XP
  • Import the sklearn Flavor from the mlflow module.
  • Set the Experiment to "Sklearn Model".
  • Use auto logging from the flavor to package your model.