Skip to content

Support for multilabel classificaiton #8

@neillbyrne

Description

@neillbyrne

Hi, thanks for assembling this project. I was wondering if there were any plans to incorporate multi-label classification. The labels API on the forward pass would be ideally changed from a single dim tensor (batch_size) of correct class indexes to a 2 dim tensor of (batch_size, n_labels) with binary indicators to remain consistent. Thanks again

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions