-
-
Notifications
You must be signed in to change notification settings - Fork 152
Open
Labels
Description
Is your feature request related to a problem? Please describe.
I see in the examples that multi-label classification isn't implemented yet. Is that still the case? And if so, is there a timeline for when this will be implemented?
Describe the solution you'd like
Many tabular problems require more than one-class classification. Since it's doable in vanilla PyTorch, I think it would be a fantastic addition to Tabular as well, since it makes training a lot more convenient and streamlined.
Describe alternatives you've considered
Wrapping a classifier in something like sklearn's MultiOutputClassifier introduces many issues such as lack of native validation set, early stopping, etc.
Additional context
Manoj-Paladugu, gitusrnm, Diyago and DominikFilipiak