Links

scikit-learn

scikit-learn is a supported framework on Truss. To package a scikit-learn model, follow the steps below or run this Google Colab notebook.

Install packages

If you're using a Jupyter notebook, add a line to install the sklearn and truss packages. Otherwise, ensure the packages are installed in your Python environment.
!pip install --upgrade sklearn truss
Truss officially supports scikit-learn version 1.0.2 or higher. Especially if you're using an online notebook environment like Google Colab or a bundle of packages like Anaconda, ensure that the version you are using is supported. If it's not, use the --upgrade flag and pip will install the most recent version.

Create an in-memory model

This is the part you want to replace with your own code. Using scikit-learn, build a machine learning model and keep it in-memory.
from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import load_iris
iris = load_iris()
data_x = iris['data']
data_y = iris['target']
model = RandomForestClassifier()
model.fit(data_x, data_y)

Create a Truss

Use the create command to package your model into a Truss.
from truss import create
tr = create(model, target_directory="sklearn_truss")
Check the target directory to see your new Truss!

Serve the model

To get a prediction from the Truss, try running:
tr.predict({"inputs": [[0, 0, 0, 0]]})
For more on running the Truss locally, see local development.