search

HanxunH / Active-Passive-Losses

[ICML2020] Normalized Loss Functions for Deep Learning with Noisy Labels
MIT License
129 stars 28 forks source link

Weighted loss #1

Closed JoJoNing25 closed 2 years ago

JoJoNing25 commented 2 years ago

Hi there,

Thank you for your fantastic work. It seems like none of the loss functions has the weight argument. Is it possible to apply the weight balanced loss for some unbalanced noisy datasets?

Kind regards,

Yuning Zhou

HanxunH commented 2 years ago

Feels free to add pull requests for your weighted implementation. :smiley: