Skip to content

Commit 4694631

Browse files
committed
Included calibration metrics into results summary
1 parent f384f74 commit 4694631

1 file changed

Lines changed: 12 additions & 1 deletion

File tree

src/thunder/utils/results.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,18 @@
11
import logging
22
import os
33

4-
METRICS = ["accuracy", "balanced_accuracy", "f1", "jaccard", "roc_auc"]
4+
METRICS = [
5+
"accuracy",
6+
"balanced_accuracy",
7+
"f1",
8+
"jaccard",
9+
"roc_auc",
10+
"ECE",
11+
"MCE",
12+
"SCE",
13+
"ACE",
14+
"TACE",
15+
]
516

617
TASK2CATS = {
718
"adversarial_attack": ["adversarial", "clean", "drop"],

0 commit comments

Comments
 (0)