Browse Source

fix bug of non-hpo

tags/v0.3.1
Frozenmad 5 years ago
parent
commit
e477663d35
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      autogl/solver/classifier/node_classifier.py

+ 1
- 1
autogl/solver/classifier/node_classifier.py View File

@@ -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(


Loading…
Cancel
Save