Skip to content

Instantly share code, notes, and snippets.

@aswalin

aswalin/catboost

Last active April 14, 2019 03:37
Show Gist options
  • Save aswalin/0115b045b98db0ccd01e760840dc4638 to your computer and use it in GitHub Desktop.
Save aswalin/0115b045b98db0ccd01e760840dc4638 to your computer and use it in GitHub Desktop.
Download ZIP
catboost
Raw
catboost
from catboost import *
train_data = [["a", 1, 1], [ "b", 3, 0], [ "a", 3, 1]]
test_data = [[ "a", 1, 2]]
train_labels = [10, 20, 30]
model = CatBoostRegressor(iterations=10)
model.fit(train_data, train_labels)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment