ml_grid.util.param_space

Defines parameter space constants for hyperparameter tuning.

Classes

ParamSpace

Initializes the ParamSpace with a specified size.

Module Contents

class ml_grid.util.param_space.ParamSpace(size: str | None)[source]

Initializes the ParamSpace with a specified size.

Parameters:

size (Optional[str]) – The size of the parameter space to generate. Expected values are “medium”, “xsmall”, or “xwide”. If an unrecognized size is provided, self.param_dict will be None.

param_dict: Dict[str, Any] | None = None[source]