peal.training.criterions.focal_criterion¶

peal.training.criterions.focal_criterion(model, y_pred, y_target, latent_code=None)[source]¶

Focal loss on every output column.

alpha/gamma come from model.config.task.focal_alpha and focal_gamma.