LightGBM
LightGBM is a supported framework on Truss. To package a LightGBM model, follow the steps below or run this Google Colab notebook.
If you're using a Jupyter notebook, add a line to install the
lightgbm and truss packages. Otherwise, ensure the packages are installed in your Python environment.!pip install --upgrade lightgbm truss
Truss officially supports
lightgbm version 3.3.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.This is the part you want to replace with your own code. Using LightGBM, build a machine learning model and keep it in-memory.
import lightgbm as lgb
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 = lgb.Dataset(X_train, y_train)
test = lgb.Dataset(X_test, y_test)
return train, test
train, test = create_data()
params = {
'boosting_type': 'gbdt',
'objective': 'softmax',
'metric': 'multi_logloss',
'num_leaves': 31,
'num_classes': 2,
'learning_rate': 0.05,
'feature_fraction': 0.9,
'bagging_fraction': 0.8,
'bagging_freq': 5,
'verbose': 0
}
model = lgb.train(params=params, train_set=train, valid_sets=test)
Use the
create command to package your model into a Truss.from truss import create
tr = create(model, target_directory="lightgbm_truss")
Check the target directory to see your new Truss!
To get a prediction from the Truss, try running:
tr.predict([[0, 0, 0, 0, 0, 0]])
Last modified 22d ago