Simple play icon Course
Skills Expanded

Managing Models Using MLflow on Databricks

by Janani Ravi

This course will teach you how to manage the end-to-end lifecycle of your machine learning models using the MLflow managed service on Databricks.

What you'll learn

The machine learning workflow involves many intricate steps to ensure that the model that you deploy to production is meaningful and robust. Managing this workflow manually is hard which is why the MLflow service which manages the integrated machine learning workflow end-to-end is game changing. Databricks makes this even easier by offering a managed version of this service that is simple, intuitive, and easy to use.

In this course, Managing Models Using MLflow on Databricks, you will learn to create an MLflow experiment and use it to track the runs of your models.

First, you will see how you can use explicit logging to log model-related metrics and parameters and view, sort, and compare runs in an experiment.

Next, you will then see how you can use autologging to track all relevant parameter, metrics, and artifacts without you having to explicitly write logging code.

Then, you will see how you can use MLflow to productionize and serve your models, and register your models in the model registry and perform batch inference using your model.

After that, you will learn how to transition your model through lifecycle stages such as Staging, Production, and Archived.

Finally, you will see how you can work with custom models in MLflow. You will also learn how to package your model in a reusable format as an MLflow project and run training using that project hosted on Github or on the Databricks file system.

When you are finished with this course, you will have the skills and knowledge to use MLflow on Databricks to manage the entire lifecycle of your machine learning model.

Table of contents

About the author

Janani has a Masters degree from Stanford and worked for 7+ years at Google. She was one of the original engineers on Google Docs and holds 4 patents for its real-time collaborative editing framework. After spending years working in tech in the Bay Area, New York, and Singapore at companies such as Microsoft, Google, and Flipkart, Janani finally decided to combine her love for technology with her passion for teaching. She is now the co-founder of Loonycorn, a content studio focused on providing ... more

Ready to upskill? Get started