diff --git a/pytorch_tabnet/multiclass_utils.py b/pytorch_tabnet/multiclass_utils.py index 2f4b44aa..0a0a26a7 100644 --- a/pytorch_tabnet/multiclass_utils.py +++ b/pytorch_tabnet/multiclass_utils.py @@ -67,7 +67,11 @@ def _unique_indicator(y): """ Not implemented """ - pass + raise IndexError( + f"""Given labels are of size {y.shape} while they should be (n_samples,) \n""" + + """If attempting multilabel classification, try using TabNetMultiTaskClassification """ + + """or TabNetRegressor""" + ) _FN_UNIQUE_LABELS = {