From 5f8573b1a6f579ae95c918bbd2a44dacf0db2e37 Mon Sep 17 00:00:00 2001 From: rusty1s Date: Sat, 17 Oct 2020 18:07:09 +0200 Subject: [PATCH] fix test --- train_hiv.py | 2 +- train_tox21.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/train_hiv.py b/train_hiv.py index 995b2d0..5845008 100644 --- 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 index 412e2b8..a2cc9f2 100644 --- 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 = []