ja-thomas / autoxgboost

autoxgboost - Automatic tuning and fitting of xgboost
Other
121 stars 19 forks source link

class weights #65

Open p-schaefer opened 4 years ago

p-schaefer commented 4 years ago

I left a note here: #35, but thought it might warrant its own issue (if this package is still being developed). scale_pos_weight is only for binary classification, is there an alternative for dealing with class imbalance in multiclass classifications? I don't know mlr well, but perhaps xgboost.DMatrix(..., weight = *weight array for individual weights*), or passing .weights to trainLearner.classif.xgboost, or perhaps there is a better alternative?