probly.train.calibration.torch¶
Collection of torch calibration training functions.
Classes¶
Expected Calibration Error (ECE) [GPSW17b]. |
|
Focal Loss based on [LGG+17]. |
|
Label Relaxation Loss from [LHullermeier21]. |
|
Cross-entropy loss with label smoothing. |