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

[Tune] Added method to integrate previous analysis in BO (#8486)

Browse Source
This commit is contained in:
Luca Cappelletti
2020-05-19 23:26:43 -07:00
committed by GitHub
parent f8f7efc24f
commit c9898eff24
2 changed files with 46 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/tests/test_tune_restore.py
+8 -5
View File
@@ -202,19 +202,22 @@ class HyperoptWarmStartTest(AbstractWarmStartTest, unittest.TestCase):
class BayesoptWarmStartTest(AbstractWarmStartTest, unittest.TestCase):
def set_basic_conf(self):
def set_basic_conf(self, analysis=None):
space = {"width": (0, 20), "height": (-100, 100)}
def cost(space, reporter):
reporter(loss=(space["height"] - 14)**2 - abs(space["width"] - 3))
search_alg = BayesOptSearch(
space,
metric="loss",
mode="min",
)
space, metric="loss", mode="min", analysis=analysis)
return search_alg, cost
def testBootStrapAnalysis(self):
analysis = self.run_exp_3()
search_alg3, cost = self.set_basic_conf(analysis)
search_alg3 = ConcurrencyLimiter(search_alg3, 1)
tune.run(cost, num_samples=10, search_alg=search_alg3, verbose=0)
class SkoptWarmStartTest(AbstractWarmStartTest, unittest.TestCase):
def set_basic_conf(self):