Example Projects

Here are the official BentoML example projects that you can find in the bentoml/gallery repository, grouped by the main ML training framework used in the project.

You can download the examples below and run them on your computer. Links to run them on Google Colab are also available, although some of the features demo’d in the notebooks does not work in the Colab environment due to its limitations, including running the BentoML API model server, building docker image or creating cloud deployment.

Scikit-Learn

• Sentiment Analysis - Google Colab | nbviewer | source

• Iris Classifier with Custom WebUI - Google Colab | nbviewer | source

• Multiple models with JsonInput - Google Colab | nbviewer | source

PyTorch

• Fashion MNIST - Google Colab | nbviewer | source

• CIFAR-10 Image Classification - Google Colab | nbviewer | source

Tensorflow Keras

• Fashion MNIST - Google Colab | nbviewer | source

• Text Classification - Google Colab | nbviewer | source

• Toxic Comment Classifier - Google Colab | nbviewer | source

Tensorflow 2.0

• tf.Function echo model - Google Colab | nbviewer | source

• Tf2 fashion MNIST - Google Colab | nbviewer | source

• Movie Review Sentiment with BERT - Google Colab | nbviewer | source

XGBoost

• Titanic Survival Prediction - Google Colab | nbviewer | source

• League of Legend win Prediction - Google Colab | nbviewer | source

FastAI

• Pet Image Classification - Google Colab | nbviewer | source

• Salary Range Prediction - Google Colab | nbviewer | source

• FastAI 2 Medical Image Classification Google Colab | nbviewer | Source

LightGBM

• Titanic Survival Prediction - Google Colab | nbviewer | source

H2O

• Loan Default Prediction - Google Colab | nbviewer | source

• Prostate Cancer Prediction - Google Colab | nbviewer | source

FastText

• Text Classification - Google Colab | nbviewer | source

ONNX

• Scikit-learn Iris Classifier - Google Colab | nbviewer | source

• ResNet50 Image Recognition (ONNX model zoo) - Google Colab | nbviewer | source

PaddlePaddle

• LinearRegression - Google Colab | nbviewer | Source

PaddleHub

• Image Segmentation - Google Colab | nbviewer | Source

Spacy

• Named Entity Recognizer - Google Colab | nbviewer | source

Statsmodels

• Shampoo Sales Prediction - Google Colab | nbviewer | source

AWS SageMaker

• Movie Review with TensorFlow 2.0 and BERT - Google Colab | nbviewer | source

SQL Server Machine Learning Services

• Iris Classifier deployed with SQL Server - nbviewer | source

MLflow

• Load Mlflow model - source

If you have a project built with BentoML and would like to share with the community, feel free to create a Pull Request and add your project to the list above.