xgb for multiclass requires classes to be integers enumerated from 0

uclamii / model_tuner

A library to tune the hyperparameters of common ML models. Supports calibration and custom pipelines.

Apache License 2.0

3 stars 0 forks source link

Open panas89 opened 4 days ago

panas89 commented 4 days ago

xgboost library expects class to be integer from 0
sklearn and catboost does not expect this, works with all types of class
Maybe explain in documentation that class variable has to be encoded or provide this functionality