from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split
X, y = make_classification(n_samples=100,
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)
train, test = create_data()
'metric': 'multi_logloss',
model = lgb.train(params=params, train_set=train, valid_sets=test)
baseten.login("*** INSERT API KEY ***") # https://docs.baseten.co/settings/api-keys
baseten_model = baseten.deploy(