README.md

Train and Deploy Machine Learning Models on Kubernetes with Kubeflow and Seldon-Core

Using:

• kubeflow
• seldon-core

The example will be the MNIST handwritten digit classification task. We will train 3 different models to solve this task:

• A TensorFlow neural network model.
• A scikit-learn random forest model.
• An R least squares model.

We will then show various rolling deployments

1. Deploy the single Tensorflow model.
2. Do a rolling update to an AB test of the Tensorflow model and the sklearn model.
3. Do a rolling update to a Multi-armed Bandit over all 3 models to direct traffic in real time to the best model.

In the follow we will:

1. Install kubeflow and seldon-core on a kubernetes cluster
2. Train the models
3. Serve the models

Setup

Either :

1. Follow the kubeflow docs to
   1. Create a persistent disk for NFS. Call it nfs-1.
   2. Install kubeflow with an NFS volume, Argo and seldon-core onto your cluster.
2. Follow a consolidated guide to do the steps in 1.

MNIST models

Tensorflow Model

• Python training code
• Python runtime prediction code
• Script to create wrap runtime prediction code to run under seldon-Core

SKLearn Model

• Python training code
• Python runtime prediction code
• Script to create wrap runtime prediction code to run under seldon-Core

R Model

• R training code
• R runtime prediction code
• Script to create wrap runtime prediction code to run under seldon-Core

Train the Models

Follow the steps in ./notebooks/training.ipynb to:

• Run Argo Jobs for each model to:
  • Creating training images and push to repo
  • Run training
  • Create runtime prediction images and push to repo
  • Deploy individual runtime model

Serve the Models

Follow the steps in ./notebooks/serving.ipynb to:

1. Deploy the single Tensorflow model.
2. Do a rolling update to an AB test of the Tensorflow model and the sklearn model.
3. Do a rolling update to a Multi-armed Bandit over all 3 models to direct traffic in real time to the best model.

If you have installed the Seldon-Core analytics you can view them on the grafana dashboard: