From d356ebdbd630914145787f4eb312be6c04af607b Mon Sep 17 00:00:00 2001 From: cluster32 Date: Mon, 21 Dec 2020 15:14:03 +0800 Subject: [PATCH] update default AutoNE para --- autogl/module/hpo/autone.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/autogl/module/hpo/autone.py b/autogl/module/hpo/autone.py index 7889fa0..d254a19 100644 --- a/autogl/module/hpo/autone.py +++ b/autogl/module/hpo/autone.py @@ -45,9 +45,9 @@ class AutoNE(BaseHPOptimizer): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) - self.max_evals = kwargs.get("max_evals", 3) - self.subgraphs = kwargs.get("subgraphs", 2) - self.sub_evals = kwargs.get("sub_evals", 2) + self.max_evals = kwargs.get("max_evals", 100) + self.subgraphs = kwargs.get("subgraphs", 5) + self.sub_evals = kwargs.get("sub_evals", 5) self.sample_batch_size = kwargs.get("sample_batch_size", 150) self.sample_walk_length = kwargs.get("sample_walk_length", 2)