#ISP Fix introspection signal computation and add more derived signals

PiperOrigin-RevId: 471400552
google · Sep 1, 2022 · 55a203d · 55a203d
1 parent aef38eb
commit 55a203d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/uncertainty_baselines/datasets/toxic_comments.py b/uncertainty_baselines/datasets/toxic_comments.py
@@ -60,7 +60,7 @@
 _IS_TRAIN_NAME = 'is_train'
 _SIGNAL_NAMES = [
     _IS_TRAIN_NAME, 'pseudo_labels', 'use_pseudo_label', 'bias_rank',
-    'uncertainty', 'bias', 'noise', 'variance', 'error'
+    'uncertainty', 'bias', 'noise', 'variance', 'error', 'error_rank',
 ]