Harshit Mehta
Harshit Mehta

Reputation: 338

for Imbalanced data dealing with cat boost

Is there a parameter like "scale_pos_weight" in catboost package as we used to have in the xgboost package in python ?

Upvotes: 11

Views: 21678

Answers (3)

from_mars
from_mars

Reputation: 41

Yes, you can use "auto_class_weights" and "scale_pos_weight" as described in original catboost documentations: https://catboost.ai/docs/concepts/python-reference_parameters-list.html

Upvotes: 3

Anna Veronika Dorogush
Anna Veronika Dorogush

Reputation: 1223

CatBoost also has scale_pos_weight parameter starting from version 0.6.1

Upvotes: 8

Gaarv
Gaarv

Reputation: 824

Yes, the parameter is named "class_weights", you can find it here : Documentation

You have to pass a list like [0.8, 0.2] for binary or [0.3, 0.8, 0.4, 0.6] for multiclass of 4 for example. Doesn't have to sum to 1, it's used as a multiplier.

Upvotes: 11

Related Questions