From 6d7c3d8ca25c698f4d8ea27236e9ef5e00c8d2ac Mon Sep 17 00:00:00 2001 From: lihy96 Date: Wed, 13 Oct 2021 10:52:22 +0800 Subject: [PATCH] Update graph_classification_full.py --- autogl/module/train/graph_classification_full.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/autogl/module/train/graph_classification_full.py b/autogl/module/train/graph_classification_full.py index bba8c9a..6cdf7f5 100644 --- a/autogl/module/train/graph_classification_full.py +++ b/autogl/module/train/graph_classification_full.py @@ -510,7 +510,6 @@ class GraphClassificationFullTrainer(BaseGraphClassificationTrainer): y_pred_prob = self._predict_proba(loader=loader) y_pred = y_pred_prob.max(1)[1] - # print(y_pred_prob, y_pred) y_true_tmp = [] for data in loader: @@ -521,10 +520,6 @@ class GraphClassificationFullTrainer(BaseGraphClassificationTrainer): y_true_tmp.append(labels) y_true = torch.cat(y_true_tmp, 0) - - print(y_pred, y_true) - - if not isinstance(feval, list): feval = [feval] return_signle = True