MLflow setup and basics in MLOps - Mini Project: Build & Apply

MLflow Setup and Basics

📖 Scenario: You are a data scientist starting a new machine learning project. You want to keep track of your experiments, parameters, and results easily. MLflow is a tool that helps you do this by logging your work in a simple way.

🎯 Goal: Set up MLflow in a Python script, log parameters and metrics for a simple experiment, and display the logged information.

📋 What You'll Learn

Create a Python dictionary called params with exact keys and values

Create a variable called metric_value with a specific float value

Use MLflow to start a run and log the parameters and metric

Print the logged parameters and metric values

💡 Why This Matters

🌍 Real World

MLflow helps data scientists keep track of their machine learning experiments easily, so they can compare results and reproduce work.

💼 Career

Knowing how to use MLflow is valuable for roles in machine learning engineering and data science, as it improves collaboration and experiment management.

1

Create experiment parameters

Create a Python dictionary called params with these exact entries: 'learning_rate': 0.01, 'num_trees': 100, and 'max_depth': 5.

MLOps

# Your code here

Hint

Use curly braces to create a dictionary and separate key-value pairs with commas.

2

Set a metric value

Create a variable called metric_value and set it to the float 0.85.

MLOps

params = {'learning_rate': 0.01, 'num_trees': 100, 'max_depth': 5}
# Your code here

Hint

Assign the float value directly to the variable.

3

Log parameters and metric with MLflow

Import mlflow. Use mlflow.start_run() as a context manager. Inside it, log the parameters from params using mlflow.log_params(params) and log the metric metric_value with the name 'accuracy' using mlflow.log_metric('accuracy', metric_value).

MLOps

params = {'learning_rate': 0.01, 'num_trees': 100, 'max_depth': 5}
metric_value = 0.85
# Your code here

Hint

Use with mlflow.start_run(): to start logging, then call the logging functions inside.

4

Print logged parameters and metric

Print the params dictionary and the string 'Accuracy:' followed by the metric_value variable on the next line.

MLOps

import mlflow

params = {'learning_rate': 0.01, 'num_trees': 100, 'max_depth': 5}
metric_value = 0.85

with mlflow.start_run():
    mlflow.log_params(params)
    mlflow.log_metric('accuracy', metric_value)
# Your code here

Hint

Use two print statements: one for the dictionary and one for the metric with label.

Practice

1. What is the primary purpose of MLflow in machine learning projects?

easy

A. To deploy machine learning models to mobile devices

B. To write machine learning algorithms from scratch

C. To create datasets for training models

D. To track and organize machine learning experiments

Solution

1. Step 1: Understand MLflow's role

   MLflow is designed to help manage and track experiments, not to build models or datasets.

2. Step 2: Identify the correct purpose

   Tracking and organizing experiments is the core feature of MLflow.

3. Final Answer:

   To track and organize machine learning experiments -> Option D

4. Quick Check:

   MLflow = experiment tracking [OK]

Hint: Remember MLflow tracks experiments, not builds models [OK]

Common Mistakes:

• Confusing MLflow with model building libraries
• Thinking MLflow creates datasets
• Assuming MLflow deploys models directly

2. Which command correctly installs MLflow using pip?

easy

A. pip install mlflow

B. pip get mlflow

C. install mlflow pip

D. pip mlflow install

Solution

1. Step 1: Recall pip install syntax

   The correct syntax to install a package is 'pip install package_name'.

2. Step 2: Match the command

   Only 'pip install mlflow' matches the correct syntax.

3. Final Answer:

   pip install mlflow -> Option A

4. Quick Check:

   pip install + package = correct [OK]

Hint: Use 'pip install' followed by package name [OK]

Common Mistakes:

• Using incorrect order of words
• Using 'pip get' instead of 'pip install'
• Omitting 'install' keyword

3. What happens when you run the command mlflow ui in your terminal?

medium

A. It starts a web interface to view and compare ML experiments

B. It installs MLflow on your system

C. It runs your machine learning model training

D. It deletes all previous MLflow experiments

Solution

1. Step 1: Understand the 'mlflow ui' command

   This command launches the MLflow tracking server's web interface.

2. Step 2: Identify the effect

   The UI lets users view and compare experiments visually in a browser.

3. Final Answer:

   It starts a web interface to view and compare ML experiments -> Option A

4. Quick Check:

   mlflow ui = launch web UI [OK]

Hint: Think 'ui' means user interface for experiments [OK]

Common Mistakes:

• Confusing UI launch with installation
• Assuming it runs training automatically
• Thinking it deletes experiments

4. You try to start MLflow UI with mlflow ui but get an error saying 'command not found'. What is the most likely cause?

medium

A. You need to run 'mlflow start' instead

B. Your Python version is too new for MLflow

C. MLflow is not installed or not in your system PATH

D. You must run the command inside a Jupyter notebook

Solution

1. Step 1: Analyze the error message

   'command not found' means the system cannot locate the 'mlflow' command.

2. Step 2: Identify common causes

   This usually happens if MLflow is not installed or its executable is not in the system PATH.

3. Final Answer:

   MLflow is not installed or not in your system PATH -> Option C

4. Quick Check:

   Command not found = missing install or PATH [OK]

Hint: Check if MLflow is installed and in PATH [OK]

Common Mistakes:

• Trying wrong commands like 'mlflow start'
• Blaming Python version without checking install
• Assuming it must run inside Jupyter

5. You want to create a new MLflow experiment named 'MyExperiment' and log a parameter 'alpha' with value 0.5 in a Python script. Which code snippet correctly does this?

hard

A. import mlflow
mlflow.create_experiment('MyExperiment')
mlflow.log_param('alpha', 0.5)

B. import mlflow
mlflow.set_experiment('MyExperiment')
with mlflow.start_run():
mlflow.log_param('alpha', 0.5)

C. import mlflow
mlflow.start_experiment('MyExperiment')
mlflow.log_param('alpha', 0.5)

D. import mlflow
mlflow.set_experiment('MyExperiment')
mlflow.log_param('alpha', 0.5)

Solution

1. Step 1: Set the experiment name

   Use mlflow.set_experiment('MyExperiment') to select or create the experiment.

2. Step 2: Start a run and log parameters

   Use 'with mlflow.start_run():' block to start a run, then log parameters inside it.

3. Step 3: Identify correct snippet

   import mlflow mlflow.set_experiment('MyExperiment') with mlflow.start_run(): mlflow.log_param('alpha', 0.5) correctly uses set_experiment, start_run context, and logs parameter.

4. Final Answer:

   import mlflow mlflow.set_experiment('MyExperiment') with mlflow.start_run(): mlflow.log_param('alpha', 0.5) -> Option B

5. Quick Check:

   Set experiment + start run + log param = correct [OK]

Hint: Always start a run before logging parameters [OK]

Common Mistakes:

• Logging parameters outside a run
• Using non-existent functions like create_experiment
• Not using 'with' block for start_run