include a mask for ece calculation

google-research / robustness_metrics

Apache License 2.0

463 stars 33 forks source link

Open ekellbuch opened 1 year ago

ekellbuch commented 1 year ago

This code includes a sample_weight variable to mask the logits used to calculate the ECE.