Skip to content

Multi-label classification #70

@dettmar

Description

@dettmar

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

Metadata

Metadata

Assignees

No one assigned

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions