Browse Source

update default AutoNE para

tags/v0.1.1^2
cluster32 5 years ago
parent
commit
d356ebdbd6
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      autogl/module/hpo/autone.py

+ 3
- 3
autogl/module/hpo/autone.py View File

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



Loading…
Cancel
Save