setu.MixedNLESearchSpace

class setu.MixedNLESearchSpace(flow_type=('nsf', 'maf'), n_layers=(2, 12), nn_width=(32, 64, 128, 256), nn_depth=(1, 3), activation=('relu', 'tanh', 'silu'), learning_rate=(0.0001, 0.005), knots=(4, 16), interval=(2.0, 6.0), n_flow_layers=(2, 10), n_made_layers=(1, 4), made_hidden_features=(32, 64, 128))

Bases: NLESearchSpace

Search space for MixedNLE hyperparameter tuning.

Inherits flow params from NLESearchSpace and adds MADE (discrete component) params. Overrides n_layers default to (2, 10) as n_flow_layers.

Parameters:

• flow_type (tuple)

• n_layers (tuple[int, int])

• nn_width (tuple)

• nn_depth (tuple[int, int])

• activation (tuple)

• learning_rate (tuple[float, float])

• knots (tuple[int, int])

• interval (tuple[float, float])

• n_flow_layers (tuple[int, int])

• n_made_layers (tuple[int, int])

• made_hidden_features (tuple)