Module neuralops.config.sweep_config

Classes

class SweepConfig (name: str = 'sweep',
algorithm: str = 'tpe',
direction: str = 'maximize',
metric: str = 'eval_accuracy',
max_trials: int = 20,
parallel_workers: int = 1,
pruner: str = 'hyperband',
search_space: dict[str, typing.Any] = <factory>,
base_config: dict[str, typing.Any] = <factory>)

Expand source code

@dataclass
class SweepConfig:
    name: str = "sweep"
    algorithm: str = "tpe"  # tpe, random, grid, cmaes
    direction: str = "maximize"
    metric: str = "eval_accuracy"
    max_trials: int = 20
    parallel_workers: int = 1
    pruner: str = "hyperband"
    search_space: dict[str, Any] = field(default_factory=dict)
    base_config: dict[str, Any] = field(default_factory=dict)

SweepConfig(name: str = 'sweep', algorithm: str = 'tpe', direction: str = 'maximize', metric: str = 'eval_accuracy', max_trials: int = 20, parallel_workers: int = 1, pruner: str = 'hyperband', search_space: dict[str, typing.Any] = , base_config: dict[str, typing.Any] = )

Instance variables

var algorithm : str

The type of the None singleton.

var base_config : dict[str, typing.Any]

The type of the None singleton.

var direction : str

The type of the None singleton.

var max_trials : int

The type of the None singleton.

var metric : str

The type of the None singleton.

var name : str

The type of the None singleton.

var parallel_workers : int

The type of the None singleton.

var pruner : str

The type of the None singleton.

var search_space : dict[str, typing.Any]

The type of the None singleton.