yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-12-25 22:18:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

use cfo in tune.run if bs is not installed (#334)

Browse Source 
Use cfo in tune.run if bs is not installed and issue a warning.
This commit is contained in:
Chi Wang 2021-12-12 21:10:06 -08:00 committed by GitHub
parent 434586e2e2
commit 228b00cbee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
flaml/tune/tune.py
View File

@ -308,7 +308,7 @@ def run(
else:
logger.setLevel(logging.CRITICAL)
from ..searcher.blendsearch import BlendSearch
from ..searcher.blendsearch import BlendSearch, CFO
if search_alg is None:
flaml_scheduler_resource_attr = (
@ -325,7 +325,17 @@ def run(
flaml_scheduler_max_resource = max_resource
flaml_scheduler_reduction_factor = reduction_factor
scheduler = None
search_alg = BlendSearch(
try:
import optuna
SearchAlgorithm = BlendSearch
except ImportError:
SearchAlgorithm = CFO
logger.warning(
"Using CFO for search. To use BlendSearch, run: pip install flaml[blendsearch]"
)
search_alg = SearchAlgorithm(
metric=metric or DEFAULT_METRIC,
mode=mode,
space=config,