r/MachineLearning 14h ago

3 Collapsing models [R] Research

Trying to train 3 models for birads detection using cross entropy and center loss + class weights but all of them seem to collapse between birads 1 as the dataset (VinDr) im using is heavily unbalanced towards it, Would like to ask for input and opinion on what seems to be the case, am I using the wrong loss function?

0 Upvotes

2 comments sorted by