Simple play icon Course

Building Machine Learning Models on Databricks

by Janani Ravi

This course will teach you how you can build and train your traditional machine learning models using the Databricks Machine Learning runtime and MLflow to manage the end-to-end machine learning lifecycle.

What you'll learn

Training, evaluating, and deploying machine learning models are now routine in many organizations, and having the right environment around this process is often what sets a company apart from its competitors. The Databricks Machine Learning Runtime, along with MLFlow, manages your experiment's runs, and models make training and hyperparameter tuning of your models simple and intuitive.

In this course, Building Machine Learning Models on Databricks, you will learn to build and train regression and classification models using the scikit-learn framework.

First, you will load, explore, and process your data using Databricks notebooks and you will use Bamboolib for no-code data analysis and transformations.

Next, you will create experiments and track your model’s parameters and metrics using runs, and compare runs using the MLflow UI.

After that, you will build and train regression and classification models using gradient boosting algorithms which are part of the XGBoost framework. You will also productionize and serve your models using Classic MLFlow Model Serving and perform real-time inference using your deployed models.

Finally, you will learn how you can use the Hyperopt tool for hyperparameter tuning of your models, as well as running hyperparameter tuning in a distributed fashion on a Spark cluster using the SparkTrials class.

When you are finished with this course, you will have the skills and knowledge to build and train traditional machine learning models on Databricks using MLflow to manage your machine learning workflow.

Table of contents

Course Overview

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