An interactive Jupyter Notebook that demonstrates how to leverage MLflow for experiment tracking, model logging, and lifecycle management in an end-to-end machine learning workflow.
- Project Overview
- Contents of the Notebook
- Getting Started
- Usage
- MLflow Features Used
- Technologies & Dependencies
- Authors & Contact
- License
This notebook provides a step-by-step guide for building, tracking, and managing a machine learning model using MLflow. You'll learn to:
- Train a model on a dataset
- Log parameters, metrics, and artifacts
- Visualize experiment runs using the MLflow Tracking UI
- Package and optionally register models for deployment
It’s designed to be a practical reference for experimenting with MLflow in research or production settings.
The notebook is structured into the following sections:
-
Setup & Imports
Installing necessary libraries and configuring MLflow. -
Data Loading & Preprocessing
Importing datasets and preparing features. -
Model Training & Evaluation
Training a model (e.g., regression or classification), evaluating accuracy, logging run metadata. -
MLflow Logging
Usingmlflow.start_run()to log:- Parameters (e.g., hyperparameters)
- Metrics (e.g., accuracy, RMSE)
- Artifacts (e.g., plots, model files)
- Model in MLflow model format for later inference or deployment
-
MLflow UI
How to launch and usemlflow uito compare runs and inspect logged data.
- Clone the repository
git clone https://github.com/monika393/ML-Flow.git cd ML-Flow