no "alpha" in focal_loss_softmax

fudannlp16 / focal-loss

Tensorflow version implementation of focal loss for binary and multi classification

108 stars 27 forks source link

Open DRosemei opened 6 years ago

DRosemei commented 6 years ago

Dose "alpha' make no contribution to multi_class classification?

fudannlp16 commented 6 years ago

The 'alpha' amount to the classes weights. For multi-classification, you should set up it by yourself. (default 1.0). I have removed this parameter.