-
Notifications
You must be signed in to change notification settings - Fork 310
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Summary: Pull Request resolved: #932 SAASBO is not currently supported in MBO (David and Max are working on it), reverting to the standard setup here. Reviewed By: lena-kashtelyan Differential Revision: D35853310 fbshipit-source-id: d8293ab378032e6af5059b1b477f415f25afa55c
- Loading branch information
1 parent
ca46192
commit 776f5a3
Showing
4 changed files
with
83 additions
and
39 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,51 @@ | ||
# Copyright (c) Meta Platforms, Inc. and affiliates. | ||
# | ||
# This source code is licensed under the MIT license found in the | ||
# LICENSE file in the root directory of this source tree. | ||
|
||
from ax.benchmark.benchmark_method import BenchmarkMethod | ||
from ax.modelbridge.generation_strategy import GenerationStrategy, GenerationStep | ||
from ax.modelbridge.registry import Models | ||
from ax.service.scheduler import SchedulerOptions | ||
|
||
|
||
def get_saasbo_default() -> BenchmarkMethod: | ||
generation_strategy = GenerationStrategy( | ||
name="SOBOL+FULLYBAYESIAN::default", | ||
steps=[ | ||
GenerationStep(model=Models.SOBOL, num_trials=5, min_trials_observed=3), | ||
GenerationStep( | ||
model=Models.FULLYBAYESIAN, | ||
num_trials=-1, | ||
), | ||
], | ||
) | ||
|
||
scheduler_options = SchedulerOptions(total_trials=30) | ||
|
||
return BenchmarkMethod( | ||
name=generation_strategy.name, | ||
generation_strategy=generation_strategy, | ||
scheduler_options=scheduler_options, | ||
) | ||
|
||
|
||
def get_saasbo_moo_default() -> BenchmarkMethod: | ||
generation_strategy = GenerationStrategy( | ||
name="SOBOL+FULLYBAYESIANMOO::default", | ||
steps=[ | ||
GenerationStep(model=Models.SOBOL, num_trials=5, min_trials_observed=3), | ||
GenerationStep( | ||
model=Models.FULLYBAYESIANMOO, | ||
num_trials=-1, | ||
), | ||
], | ||
) | ||
|
||
scheduler_options = SchedulerOptions(total_trials=30) | ||
|
||
return BenchmarkMethod( | ||
name=generation_strategy.name, | ||
generation_strategy=generation_strategy, | ||
scheduler_options=scheduler_options, | ||
) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters