tabensemb.model.TorchModel._space#

method

TorchModel._space(model_name)[source]#

A list of scikit-optimize search spaces for the selected model. It should contain all parameters defined in _initial_values().

Parameters:
model_name:

The name of a selected model that is currently going through Bayesian optimization.

Returns:
list

A list of skopt.space.