fix from_logits param missed by from_config of AUC metrics
@@ -1992,7 +1992,8 @@ class AUC(base_metric.Metric):
'curve': self.curve.value,
'summation_method': self.summation_method.value,
'multi_label': self.multi_label,
- 'label_weights': label_weights
+ 'label_weights': label_weights,
+ 'from_logits': self._from_logits
}
# optimization to avoid serializing a large number of generated thresholds
if self._init_from_thresholds:
