Skip to content

Multi-Target Classification #430

@YonyBresler

Description

@YonyBresler

Is your feature request related to a problem? Please describe.
Currently, regression allows for multi-target training, while classification only supports a single target.

Describe the solution you'd like
Add support for multi-target classification, including data handling, output layer size shaping, and appropriate loss.

Describe alternatives you've considered
You currently need to train multiple models, which is both slower and may not get the same performance as learning a common embedding

Additional context
I'm prepared to contribute this feature myself, but wanted to check:

  1. That there is no unworkable/undesirable roadblock that prevents this from being possible at this time
  2. That if I were to complete it, that the community is interested in integrating this feature

Metadata

Metadata

Assignees

No one assigned

    Labels

    wontfixThis will not be worked on

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions