wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-26 13:37:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] Update ZOOpt to better support the latest Ray (#11462)

Browse Source 
Co-authored-by: Servon <zewen.li@polixir.ai>
This commit is contained in:
Servon
2020-10-19 09:28:11 -07:00
committed by GitHub
co-authored by Servon
parent 798bd6a359
commit 202b1859ef
4 changed files with 56 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/tests/test_sample.py
+14 -7
View File
@@ -533,7 +533,7 @@ class SearchSpaceTest(unittest.TestCase):
"a": tune.sample.Categorical([2, 3, 4]).uniform(),
"b": {
"x": tune.sample.Integer(0, 5).quantized(2),
"y": 4,
"y": tune.sample.Categorical([2, 4, 6, 8]).uniform(),
"z": tune.sample.Float(1e-4, 1e-2).loguniform()
}
}
@@ -544,7 +544,7 @@ class SearchSpaceTest(unittest.TestCase):
"a": 2,
"b": {
"x": tune.sample.Integer(0, 5).uniform(),
"y": 4,
"y": tune.sample.Categorical([2, 4, 6, 8]).uniform(),
"z": tune.sample.Float(-3, 7).uniform().quantized(1e-4)
}
}
@@ -552,11 +552,16 @@ class SearchSpaceTest(unittest.TestCase):
zoopt_config = {
"b/x": (ValueType.DISCRETE, [0, 5], True),
"b/z": (ValueType.CONTINUOUS, [-3, 7], 1e-4)
"b/y": (ValueType.GRID, [2, 4, 6, 8]),
"b/z": (ValueType.CONTINUOUS, [-3, 7], 1e-4),
}
searcher1 = ZOOptSearch(dim_dict=converted_config, budget=5)
searcher2 = ZOOptSearch(dim_dict=zoopt_config, budget=5)
zoopt_search_config = {"parallel_num": 4}
searcher1 = ZOOptSearch(
dim_dict=converted_config, budget=5, **zoopt_search_config)
searcher2 = ZOOptSearch(
dim_dict=zoopt_config, budget=5, **zoopt_search_config)
np.random.seed(1234)
config1 = searcher1.suggest("0")
@@ -565,14 +570,16 @@ class SearchSpaceTest(unittest.TestCase):
self.assertEqual(config1, config2)
self.assertIn(config1["b"]["x"], list(range(5)))
self.assertIn(config1["b"]["y"], [2, 4, 6, 8])
self.assertLess(-3, config1["b"]["z"])
self.assertLess(config1["b"]["z"], 7)
searcher = ZOOptSearch(budget=5, metric="a", mode="max")
searcher = ZOOptSearch(
budget=5, metric="a", mode="max", **zoopt_search_config)
analysis = tune.run(
_mock_objective, config=config, search_alg=searcher, num_samples=1)
trial = analysis.trials[0]
self.assertEqual(trial.config["b"]["y"], 4)
self.assertIn(trial.config["b"]["y"], [2, 4, 6, 8])
if __name__ == "__main__":