-
Notifications
You must be signed in to change notification settings - Fork 1.4k
Open
Description
Is your feature request related to a problem? Please describe.
In various datasets, background class or unlabeled class is index by some number, a simple ignore_class = IGNORE_CLASS can help.
Describe the solution you'd like
Add a ignore_class argument everywhere, which takes integer class value to be ignored in losses.
coderabbitai
Metadata
Metadata
Assignees
Labels
No labels