deephyper.search.nas.AMBSMixed#
- class deephyper.search.nas.AMBSMixed(problem, evaluator, random_state=None, log_dir='.', verbose=0, surrogate_model: str = 'RF', acq_func: str = 'UCB', kappa: float = 1.96, xi: float = 0.001, n_points: int = 10000, liar_strategy: str = 'cl_max', n_jobs: int = 1, **kwargs)[source]#
Bases:
NeuralArchitectureSearch
Asynchronous Model-Based Search baised on the Scikit-Optimized Optimizer. It is extended to the case of joint hyperparameter and neural architecture search.
- Parameters:
problem (NaProblem) – Neural architecture search problem describing the search space to explore.
evaluator (Evaluator) – An
Evaluator
instance responsible of distributing the tasks.random_state (int, optional) – Random seed. Defaults to None.
log_dir (str, optional) – Log directory where search’s results are saved. Defaults to “.”.
verbose (int, optional) – Indicate the verbosity level of the search. Defaults to 0.
surrogate_model (str, optional) – Surrogate model used by the Bayesian optimization. Can be a value in
["RF", "ET", "GBRT", "DUMMY"]
. Defaults to"RF"
.acq_func (str, optional) – Acquisition function used by the Bayesian optimization. Can be a value in
["UCB", "EI", "PI", "gp_hedge"]
. Defaults to"UCB"
.kappa (float, optional) – Manage the exploration/exploitation tradeoff for the “UCB” acquisition function. Defaults to
1.96
for a balance between exploitation and exploration.xi (float, optional) – Manage the exploration/exploitation tradeoff of
"EI"
and"PI"
acquisition function. Defaults to0.001
for a balance between exploitation and exploration.n_points (int, optional) – The number of configurations sampled from the search space to infer each batch of new evaluated configurations. Defaults to
10000
.liar_strategy (str, optional) – Definition of the constant value use for the Liar strategy. Can be a value in
["cl_min", "cl_mean", "cl_max"]
. Defaults to"cl_max"
.n_jobs (int, optional) – Number of parallel processes used to fit the surrogate model of the Bayesian optimization. A value of
-1
will use all available cores. Defaults to1
.
Methods
check_evaluator
Dumps the context in the log folder.
Execute the search algorithm.
Returns a json version of the search object.
Attributes
The identifier of the search used by the evaluator.
- dump_context()#
Dumps the context in the log folder.
- search(max_evals: int = -1, timeout: int | None = None)#
Execute the search algorithm.
- Parameters:
- Returns:
a pandas DataFrame containing the evaluations performed or
None
if the search could not evaluate any configuration.- Return type:
DataFrame
- property search_id#
The identifier of the search used by the evaluator.
- to_json()#
Returns a json version of the search object.