Skip to content

Request: Support for Multilabel #10

@vody-am

Description

@vody-am

Hi! Thanks for this cool project. I think it'd be very useful to add in support for multilabel classification as well. This might require training with BCEWithLogitsLoss, and accepting a different dataset format. I attempted to add it in but ran into some bugs. If interested, this feature would be appreciated. If I manage to get it working, I'll send a PR.

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions