From e477663d35539bc7fa9affc8c012e501c272c844 Mon Sep 17 00:00:00 2001 From: Frozenmad Date: Wed, 24 Feb 2021 14:05:55 +0000 Subject: [PATCH] fix bug of non-hpo --- autogl/solver/classifier/node_classifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/autogl/solver/classifier/node_classifier.py b/autogl/solver/classifier/node_classifier.py index 9274e1b..d663357 100644 --- a/autogl/solver/classifier/node_classifier.py +++ b/autogl/solver/classifier/node_classifier.py @@ -312,7 +312,7 @@ class AutoNodeClassifier(BaseClassifier): ) if self.hpo_module is None: model.initialize() - model.train(self.data, True) + model.train(self.dataset, True) optimized = model else: optimized, _ = self.hpo_module.optimize(