deephyper.search.nas.AMBSMixed
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:
deephyper.search.nas._base.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
Dumps the context in the log folder.
Execute the search algorithm.
Terminate the search.
Returns a json version of the search object.
-
dump_context
()# Dumps the context in the log folder.
-
search
(max_evals: int = - 1, timeout: Optional[int] = 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
-
terminate
()# Terminate the search.
- Raises
SearchTerminationError – raised when the search is terminated with SIGALARM
-
to_json
()# Returns a json version of the search object.