focal-loss icon indicating copy to clipboard operation
focal-loss copied to clipboard

Published 20 hours ago • verified publisher icon artemmavrin

SparseCategoricalFocalLoss: ignore class?

Open rdisipio opened this issue 2 years ago • 0 comments

Hi,

TensorFlow's implementation of categorical cross-entropy accepts a parameter called ignore_class (see here) since v 2.10. In principle, it can be useful for NER tasks. It would be nice if such a parameter appeared in the calculation of focal loss, so an apple-to-apple comparison can be carried out.

rdisipio avatar Oct 27 '22 15:10 rdisipio