"src/nni_manager/vscode:/vscode.git/clone" did not exist on "d7456c168231f35bd510ace6256570946ea55d83"
Unverified Commit f579f176 authored by Yuge Zhang's avatar Yuge Zhang Committed by GitHub
Browse files

Fix DNGO tuner class name (#3707)

parent 969fac0d
...@@ -40,7 +40,7 @@ def _random_config(search_space, random_state): ...@@ -40,7 +40,7 @@ def _random_config(search_space, random_state):
return chosen_config return chosen_config
class DngoTuner(Tuner): class DNGOTuner(Tuner):
def __init__(self, optimize_mode='maximize', sample_size=1000, trials_per_update=20, num_epochs_per_training=500): def __init__(self, optimize_mode='maximize', sample_size=1000, trials_per_update=20, num_epochs_per_training=500):
self.searchspace_json = None self.searchspace_json = None
......
...@@ -12,7 +12,7 @@ from collections import deque ...@@ -12,7 +12,7 @@ from collections import deque
from unittest import TestCase, main from unittest import TestCase, main
from nni.algorithms.hpo.batch_tuner import BatchTuner from nni.algorithms.hpo.batch_tuner import BatchTuner
from nni.algorithms.hpo.dngo_tuner import DngoTuner from nni.algorithms.hpo.dngo_tuner import DNGOTuner
from nni.algorithms.hpo.evolution_tuner import EvolutionTuner from nni.algorithms.hpo.evolution_tuner import EvolutionTuner
from nni.algorithms.hpo.gp_tuner import GPTuner from nni.algorithms.hpo.gp_tuner import GPTuner
from nni.algorithms.hpo.gridsearch_tuner import GridSearchTuner from nni.algorithms.hpo.gridsearch_tuner import GridSearchTuner
...@@ -390,7 +390,7 @@ class BuiltinTunersTestCase(TestCase): ...@@ -390,7 +390,7 @@ class BuiltinTunersTestCase(TestCase):
self.import_data_test_for_pbt() self.import_data_test_for_pbt()
def test_dngo(self): def test_dngo(self):
tuner_fn = lambda: DngoTuner(trials_per_update=100, num_epochs_per_training=1) tuner_fn = lambda: DNGOTuner(trials_per_update=100, num_epochs_per_training=1)
self.search_space_test_all(tuner_fn, fail_types=["choice_str", "choice_mixed", self.search_space_test_all(tuner_fn, fail_types=["choice_str", "choice_mixed",
"normal", "lognormal", "qnormal", "qlognormal"]) "normal", "lognormal", "qnormal", "qlognormal"])
self.import_data_test(tuner_fn, stype='choice_num') self.import_data_test(tuner_fn, stype='choice_num')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment