-
Notifications
You must be signed in to change notification settings - Fork 28
Open
Description
Hi,
I have a classification task for a rating system over 5 classes. I want now to switch to ordinal regression but I am a bit lost regarding the importance weights.
So for the moment, for example, my weights are, for my 5 classes : tensor([1.4237, 1.0000, 2.4000, 1.2923, 1.1200])
(so my dominant class is 2 and the one with the least examples is the 3)
How should I convert it to pass it to importance_weights of coral_loss ?
Regards,
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels