fix test

rusty1s · Oct 17, 2020 · 5f8573b · 5f8573b
1 parent 18fef4e
commit 5f8573b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/train_hiv.py b/train_hiv.py
@@ -92,7 +92,7 @@ def test(loader):
             rocauc_list.append(
                 roc_auc_score(y_true[is_labeled, i], y_pred[is_labeled, i]))
 
-    return {"rocauc": sum(rocauc_list) / len(rocauc_list)}
+    return sum(rocauc_list) / len(rocauc_list)
 
 
 test_perfs = []

diff --git a/train_tox21.py b/train_tox21.py
@@ -92,7 +92,7 @@ def test(loader):
             rocauc_list.append(
                 roc_auc_score(y_true[is_labeled, i], y_pred[is_labeled, i]))
 
-    return {"rocauc": sum(rocauc_list) / len(rocauc_list)}
+    return sum(rocauc_list) / len(rocauc_list)
 
 
 test_perfs = []