CatBoost + openml__arrhythmia__5 error
duncanmcelfresh opened this issue · comments
error found with dataset openml__arrhythmia__5 using CatBoost (conda env: gbdt)
catboost/private/libs/target/target_converter.cpp:226: Unknown class label: \"11\"
this happens both with and without GPU
full traceback:
File "/home/shared/tabzilla/TabSurvey/tabzilla_experiment.py", line 134, in __call__
result = cross_validation(model, self.dataset, self.time_limit)
File "/home/shared/tabzilla/TabSurvey/tabzilla_utils.py", line 237, in cross_validation
loss_history, val_loss_history = curr_model.fit(
File "/home/shared/tabzilla/TabSurvey/models/tree_models.py", line 149, in fit
self.model.fit(X, y, eval_set=(X_val, y_val))
File "/opt/conda/envs/gbdt/lib/python3.9/site-packages/catboost/core.py", line 4716, in fit
self._fit(X, y, cat_features, text_features, embedding_features, None, sample_weight, None, None, None, None, baseline, use_best_model,
File "/opt/conda/envs/gbdt/lib/python3.9/site-packages/catboost/core.py", line 2037, in _fit
self._train(
File "/opt/conda/envs/gbdt/lib/python3.9/site-packages/catboost/core.py", line 1464, in _train
self._object._train(train_pool, test_pool, params, allow_clear_pool, init_model._object if init_model else None)
File "_catboost.pyx", line 4393, in _catboost._CatBoost._train
File "_catboost.pyx", line 4442, in _catboost._CatBoost._train
_catboost.CatBoostError: catboost/private/libs/target/target_converter.cpp:226: Unknown class label: "11"
this is also related to #27 ...