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

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions