frameworks.rst

Frameworks

Here are the all of the supported ML frameworks for BentoML. You can find the official BentoML example projects in the bentoml/gallery repository, group by the ML training frameworks 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

Example Projects:

• Sentiment Analysis - Google Colab | nbviewer | source

bentoml.artifact.SklearnModelArtifact

PyTorch

Example Projects:

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

bentoml.artifact.PytorchModelArtifact

Tensorflow 2.0

Example Projects:

• tf.Function model - Google Colab | nbviewer | source
• Fashion MNIST - Google Colab | nbviewer | source
• Movie Review Sentiment with BERT - Google Colab | nbviewer | source

bentoml.artifact.TensorflowSavedModelArtifact

Tensorflow Keras

Example Projects:

• Fashion MNIST - Google Colab | nbviewer | source
• Text Classification - Google Colab | nbviewer | source
• Toxic Comment Classifier - Google Colab | nbviewer | source

bentoml.artifact.KerasModelArtifact

FastAI

Example Projects:

• Pet Image Classification - Google Colab | nbviewer | source
• Salary Range Prediction - Google Colab | nbviewer | source

bentoml.artifact.FastaiModelArtifact

bentoml.adapters.FastaiImageInput

XGBoost

Example Projects:

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

bentoml.artifact.XgboostModelArtifact

LightGBM

Example Projects:

• Titanic Survival Prediction - Google Colab | nbviewer | source

bentoml.artifact.LightGBMModelArtifact

FastText

Example Projects:

• Text Classification - Google Colab | nbviewer | source

bentoml.artifact.FasttextModelArtifact

H2O

Example Projects:

• Loan Default Prediction - Google Colab | nbviewer | source
• Prostate Cancer Prediction - Google Colab | nbviewer | source

bentoml.artifact.H2oModelArtifact