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

[tune] Add Basic Variant Generator to search algorithm shim function (#11334)

Browse Source 
* Add Basic Variant Generator

* Add 'random' key to SEARCH_ALG_IMPORT

Co-authored-by: Kai Fricke <kai@anyscale.com>

Co-authored-by: Kai Fricke <kai@anyscale.com>
This commit is contained in:
Sumanth Ratna
2020-10-15 00:41:47 -04:00
committed by GitHub
parent 00e0f14c6f
commit 3fe757391b
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/suggest/__init__.py
+5
View File
@@ -31,6 +31,9 @@ def create_searcher(
>>> search_alg = tune.create_searcher('ax')
"""
def _import_variant_generator():
return BasicVariantGenerator
def _import_ax_search():
from ray.tune.suggest.ax import AxSearch
return AxSearch
@@ -72,6 +75,8 @@ def create_searcher(
return SigOptSearch
SEARCH_ALG_IMPORT = {
"variant_generator": _import_variant_generator,
"random": _import_variant_generator,
"ax": _import_ax_search,
"dragonfly": _import_dragonfly_search,
"skopt": _import_skopt_search,