jaxopt.loss.binary_sparsemax_loss
- jaxopt.loss.binary_sparsemax_loss(label, logit)[source]
Binary sparsemax loss.
- Parameters
label (
int
) – ground-truth integer label (0 or 1).logit (
float
) – score produced by the model (float).
- Return type
float
- Returns
loss value
References
Learning with Fenchel-Young Losses. Mathieu Blondel, André F. T. Martins, Vlad Niculae. JMLR 2020. (Sec. 4.4)