Skip to content

Coral_loss importance_weights #7

@DbrRoxane

Description

@DbrRoxane

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,

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions