Skip to content

Tutorial project to learn about Kedro and MLFlow usage for ML development.

Notifications You must be signed in to change notification settings

cscardoso/kedro-mlflow-tutorial

 
 

Repository files navigation

Kedro MLFlow Tutorial

This tutorial objective is to apply engineering best practices in machine learning development and experimentation using the Kedro framework along with a Kedro's plugin for MLFlow. The information in this README file is limited, as the tutorial video is available here. So if the information here is not enough for you, please check the video (which is in Portuguese).

Table of Contents

1. Prerequisites

1.1. Install Docker Engine

Follow the instructions here to install Docker Engine for you OS.

1.2. Install Git

Follow the instructions here to install Git for you OS.

1.3. Install you favorite code editor

Choose the code editor of you preference (PyCharm, Atom, VSCode, etc...) to help you following the tutorial.

2. Setup

2.1. Create a new folder to use on the tutorial

mkdir tutorial && cd tutorial

2.2. Clone the tutorial complete code

git clone https://github.com/rodra-go/kedro-mlflow-tutorial.git

2.3. Create a new folder to do the follow-trough of the tutorial

mkdir follow-trough && cd follow-trough

2.4. Inicialize Git

git init

3. Docker

3.1. Copy the Dockerfile from the complete code

cp ../kedro-mlflow-tutorial/Dockerfile-zero ./Dockerfile

3.1. Build Docker Image from Dockerfile

docker build -t kedro-mlflow-tutorial:0.1 .

3.2. Run the Docker container (Linux)

docker run --rm --name kedro_mlflow_tutorial -dit -p 4141:4141 -p 8888:8888 -p 5000:5000 -v $(pwd):/usr/src/code/ kedro-mlflow-tutorial:0.1

3.3. Run the Docker container (Windows)

Replace the $(pwd) on the command above for the path given by the command pwd on Windows Power Shell, then run the command. pwd on Windows Power Shell, then run the command.

docker run --rm --name kedro_mlflow_tutorial -dit -p 4141:4141 -p 8888:8888 -p 5000:5000 -v <C:/path/to/tutorial/follow-trough>:/usr/src/code/ kedro-mlflow-tutorial:0.1

3.4. Open a bash in the container

docker exec -it kedro_mlflow_tutorial bash

4. Kedro

4.1. Check Kedro installation

kedro info

4.2. Create a Kedro Project

kedro new

Name your project Kedro MLFlow Tutorial (otherwise the code might no work).

4.3. Move files properly

shopt -s dotglob && mv ./kedro-mlflow-tutorial/* . && rm -rf ./kedro-mlflow-tutorial

4.4. Initialize MLFlow plugin

kedro mlflow init

4.5 Copy the utils folder from the complete code

cp -R ../kedro-mlflow-tutorial/src/kedro_mlflow_tutorial/utils ./src/kedro_mlflow_tutorial/utils

4.6 Copy the conf folder from the complete code

cp -R ../kedro-mlflow-tutorial/conf ./conf

4.7 Copy the notebooks folder from the complete code

cp -R ../kedro-mlflow-tutorial/notebooks ./notebooks

4.8. Configure your credentials

In order to access the TPN-USP file server and complete the Data Integration implementation, it is necessary to configure your credentials on TPN-USP Network. Add the following lines to your ./conf/local/credentials.yml file:

tpn:
  username: <your_user_name>
  password: <your_password>

In case you don't have access to TPN-USP network, download the raw data here and extract its contents to ./data/01_raw in your follow-trough directory.

4.9. Run Jupyter Server

kedro jupyter notebook --ip 0.0.0.0

4.10. Run Kedro Viz

kedro viz --host 0.0.0.0

5. MLFlow

5.1. Run MLFlow Server

mlflow ui --host 0.0.0.0 --backend-store-uri file:///usr/src/code/mlruns

Overview

This is your new Kedro project, which was generated using Kedro 0.16.6.

Take a look at the Kedro documentation to get started.

Rules and guidelines

In order to get the best out of the template:

  • Don't remove any lines from the .gitignore file we provide
  • Make sure your results can be reproduced by following a data engineering convention
  • Don't commit data to your repository
  • Don't commit any credentials or your local configuration to your repository. Keep all your credentials and local configuration in conf/local/

How to install dependencies

Declare any dependencies in src/requirements.txt for pip installation and src/environment.yml for conda installation.

To install them, run:

kedro install

How to run your Kedro pipeline

You can run your Kedro project with:

kedro run

How to test your Kedro project

Have a look at the file src/tests/test_run.py for instructions on how to write your tests. You can run your tests as follows:

kedro test

To configure the coverage threshold, go to the .coveragerc file.

Project dependencies

To generate or update the dependency requirements for your project:

kedro build-reqs

This will copy the contents of src/requirements.txt into a new file src/requirements.in which will be used as the source for pip-compile. You can see the output of the resolution by opening src/requirements.txt.

After this, if you'd like to update your project requirements, please update src/requirements.in and re-run kedro build-reqs.

Further information about project dependencies

How to work with Kedro and notebooks

Note: Using kedro jupyter or kedro ipython to run your notebook provides these variables in scope: context, catalog, and startup_error.

Jupyter, JupyterLab, and IPython are already included in the project requirements by default, so once you have run kedro install you will not need to take any extra steps before you use them.

Jupyter

To use Jupyter notebooks in your Kedro project, you need to install Jupyter:

pip install jupyter

After installing Jupyter, you can start a local notebook server:

kedro jupyter notebook

JupyterLab

To use JupyterLab, you need to install it:

pip install jupyterlab

You can also start JupyterLab:

kedro jupyter lab

IPython

And if you want to run an IPython session:

kedro ipython

How to convert notebook cells to nodes in a Kedro project

You can move notebook code over into a Kedro project structure using a mixture of cell tagging and Kedro CLI commands.

By adding the node tag to a cell and running the command below, the cell's source code will be copied over to a Python file within src/<package_name>/nodes/:

kedro jupyter convert <filepath_to_my_notebook>

Note: The name of the Python file matches the name of the original notebook.

Alternatively, you may want to transform all your notebooks in one go. Run the following command to convert all notebook files found in the project root directory and under any of its sub-folders:

kedro jupyter convert --all

How to ignore notebook output cells in git

To automatically strip out all output cell contents before committing to git, you can run kedro activate-nbstripout. This will add a hook in .git/config which will run nbstripout before anything is committed to git.

Note: Your output cells will be retained locally.

Package your Kedro project

Further information about building project documentation and packaging your project

About

Tutorial project to learn about Kedro and MLFlow usage for ML development.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 73.7%
  • Python 25.9%
  • Dockerfile 0.4%