HLR / DomiKnowS

36 stars 5 forks source link

BNC vs BCE with logit loss #356

Closed kordjamshidi closed 1 year ago

kordjamshidi commented 2 years ago

Where is the accurate explanation of these two losses in which problems which one should be used. Can we have this in the documentation?

hfaghihi15 commented 2 years ago

BNC Loss and BCE loss are the losses defined in the PyTorch library, we have just wrapped them here. BCE: https://pytorch.org/docs/stable/generated/torch.nn.BCELoss.html CrossEntropy: https://pytorch.org/docs/stable/generated/torch.nn.CrossEntropyLoss.html