Catboostの使い方[更新中]

CatBoostの使い方

install

conda install catboost

パラメータ

パラメータについて

Pool-CatBoost用データセット

from catboost import Pool

train = Pool(X_train, y_train)
valid = Pool(X_valid, y_valid)

cat_featuresにカテゴリ変数のカラム名を指定できる。

feature importance

feature_importances = model.get_feature_importance(train)

Catboost

Catboost.Classifier

from catboost import CatBoostClassifier
model = CatBoostClassifier()

model.fit(train_pool, 
          eval_set=validate_pool,
          early_stopping_rounds=10,
          use_best_model=True)

model.predict(test)

Catboost.Regressor

パラメータ

パラメータについて

optunaを使ったパラメータチューニング

参考

catboost.ai/
catboost.ai/python-quickstart
tutorial-catboost-overview
Pythonでcatboostを使ってみる