Links

XGBoost

XGBoost is a supported framework on Truss. To package a XGBoost 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 xgboost and truss packages. Otherwise, ensure the packages are installed in your Python environment.
!pip install --upgrade xgboost truss
Truss officially supports xgboost version 1.6.1 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 XGBoost, build a machine learning model and keep it in-memory.
import xgboost as xgb
from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split
def create_data():
X, y = make_classification(n_samples=100,
n_informative=2,
n_classes=2,
n_features=6)
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.25)
train = xgb.DMatrix(X_train, y_train)
test = xgb.DMatrix(X_test, y_test)
return train, test
train, test = create_data()
params = {
"learning_rate": 0.01,
"max_depth": 3
}
# training, we set the early stopping rounds parameter
model = xgb.train(params,
train, evals=[(train, "train"), (test, "validation")],
num_boost_round=100, early_stopping_rounds=20)

Create a Truss

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

Serve the model

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