Skip to content

Commit 8f6d917

Browse files
committed
fix: scores nomenclature
1 parent 9fb2858 commit 8f6d917

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/classifier/train/train_eval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ def evaluate(loaders_dict, device, model):
106106
auc = roc_auc_score(labels, probs)
107107
except ValueError:
108108
auc = float("nan") # e.g., only one class present
109-
results[ntype] = {"acc": acc, "auc": auc, "n": len(labels)}
109+
results[ntype] = {"accuracy": acc, "roc_auc": auc, "n": len(labels)}
110110

111111
avg_loss = total_loss / max(steps, 1)
112112

0 commit comments

Comments
 (0)