Custom scikit-learn model

In this exercise you are going to create a custom model using MLflow's pyfunc flavor. Using the insurance_charges dataset, the labels must be changed from female to 0 and male to 1 for classification during training. When using the model, the strings of female or male must be returned instead of 0 or 1.

The custom model is a Classification model based on LogisticRegression and will use a Class called CustomPredict. The CustomPredict adds an additional step in the predict method that sets your labels of 0 and 1 back to female and male when the model receives input. You will be using pyfunc flavor for logging and loading your model.

Our insurance_charges dataset will be preprocessed and model will be trained using:

lr_model = LogisticRegression().fit(X_train, y_train)

The MLflow module will be imported.

This exercise is part of the course

Introduction to MLflow

Exercise instructions

Use MLflow's pyfunc flavor to log the custom model.
Set pyfunc python_model argument to use the Custom Class CustomPredict().
Load the custom model using pyfunc.

Hands-on interactive exercise

Have a go at this exercise by completing this sample code.

# Log the pyfunc model 
____.____.____(
	artifact_path="lr_pyfunc", 
    # Set model to use CustomPredict Class
	python_model=____, 
	artifacts={"lr_model": "lr_model"}
)

run = mlflow.last_active_run()
run_id = run.info.run_id

# Load the model in python_function format
loaded_model = ____.____.____(f"runs:/{run_id}/lr_pyfunc")

Edit and Run Code

This exercise is part of the course

Introduction to MLflow

AdvancedSkill Level

4.8+

Start Course for Free

In this Chapter, you will be introduced to MLflow and how it aims to assist with some difficulties of the Machine Learning lifecycle. You will be introduced to the four main concepts that make up MLflow with a main focus on MLflow Tracking. You will learn to create experiments and runs as well as how to track metrics, parameters, and artifacts. Finally, you will search MLflow programmatically to find experiment runs that fit certain criteria.

Exercise 1: What is MLflow?Exercise 2: Components of MLflow Exercise 3: MLflow experiments Exercise 4: MLflow Tracking Exercise 5: Starting a run Exercise 6: Logging a run Exercise 7: How to retrieve active run data?Exercise 8: Querying runs Exercise 9: Search runs query options Exercise 10: Searching runs

In this Chapter, you will be introduced to MLflow Models. The MLflow Models component of MLflow plays an essential role in the Model Evaluation and Model Engineering steps of the Machine Learning lifecycle. You will learn how MLflow Models standardizes the packaging of ML models as well as how to save, log and load them. You will learn how to create custom MLflow Models to provide more flexibility to your use cases as well as how to evaluate model performance. You will utilize the powerful concept of “Flavors” and finally use the MLflow command line tool for model deployment.

Exercise 1: Introduction to MLflow Models Exercise 2: Package a machine learning model Exercise 3: Storage Format Exercise 4: What's in an MLmodel file?Exercise 5: Model API Exercise 6: Saving and loading a model Exercise 7: Logging and loading a model Exercise 8: Custom models Exercise 9: Creating a custom Python Class Exercise 10: Custom scikit-learn model

Current Exercise

Exercise 11: Scikit-learn flavor and evaluation Exercise 12: Model serving Exercise 13: Serving a model Exercise 14: Score from a served model

This Chapter introduces the concept of MLflow called the Model Registry. You will be introduced to how the Model Registry is used to manage the lifecycle of ML models. You will learn how to create and search for models in the Model Registry. You then learn how to register models to the Model Registry and learn how to transition models between predefined stages. Finally, you will also learn how to deploy models from the Model Registry.

Exercise 1: Introduction to MLflow Model Registry Exercise 2: Create a Model Exercise 3: Searching Models Exercise 4: Registering Models Exercise 5: Registering existing models Exercise 6: Registering new models Exercise 7: Model stages Exercise 8: Model stages in Model Registry Exercise 9: Transitioning model stages Exercise 10: Model deployment Exercise 11: Serving models from the Model Registry Exercise 12: Loading models from the Model Registry

In this chapter, you'll gain valuable knowledge on how to streamline your data science code for reusability and reproducibility using MLflow Projects. The chapter begins by introducing the concept of MLflow Projects and walking you through creating an MLproject file. From there, you'll learn how to run MLflow Projects through both the command-line and the MLflow Projects module while also discovering the power of using parameters for added flexibility in your code. Finally, you will learn how to manage steps of the machine learning lifecycle by creating a multi-step workflow using MLflow Projects.

Exercise 1: Introduction to MLflow Projects Exercise 2: MLproject file layout Exercise 3: Creating an MLproject Exercise 4: Running MLflow Projects Exercise 5: MLflow run command Exercise 6: MLflow projects module Exercise 7: Specifying parameters Exercise 8: Adding parameters to MLproject Exercise 9: Adding parameters to project run Exercise 10: Workflows Exercise 11: Creating an MLproject for the ML Lifecycle: Model Engineering Exercise 12: Creating an MLproject for the ML Lifecycle: Model Evaluation Exercise 13: Creating a multi-step workflow: Model Engineering Exercise 14: Creating a multi-step workflow: Model Evaluation Exercise 15: Wrap-up