search

YanCote / IFT6268-simclr

Project for IFT6268
0 stars 0 forks source link

Weighted Loss implementation #29

Open YanCote opened 4 years ago

YanCote commented 4 years ago

Referring to https://arxiv.org/abs/1705.02315

implemented the W-CEL Loss to compensate for class imbalance in the MultiLabel Classification problem.

YanCote commented 4 years ago

First implementation done in yan/ft_on_xray. Still need validation (currently running on CC) Refer to: Finetuning.py: weighted_cel() Manual calculus WeightLoss_Calcul.pdf

sgaut023 commented 4 years ago

I'm presently looking at the loss. I will do some tests to verify that the loss is doing what we are expecting.