peal.training.criterions.mixed_focal_mse_criterion¶

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

Focal loss on the discrete outputs plus weighted MSE on the continuous.

Same column split as mixed_bce_mse_criterion(), but with FocalLoss (focal_alpha/focal_gamma from model.config.task) and mixing factor model.config.task.focal_mse_mix.

Returns:: Scalar loss.
Return type:: torch.Tensor