nat.data_models.optimizable#

Attributes#

Classes#

`SearchSpace`
`OptimizableMixin`

Functions#

OptimizableField(default, *[, space, merge_conflict])

Module Contents#

T#

class SearchSpace(/, **data: Any)#

Bases: pydantic.BaseModel, Generic[T]

values: collections.abc.Sequence[T] | None = None#

low: T | None = None#

high: T | None = None#

log: bool = False#

step: float | None = None#

is_prompt: bool = False#

prompt: str | None = None#

prompt_purpose: str | None = None#

model_config#: Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

validate_search_space_parameters()#: Validate that either values is provided, or both high and low.

suggest(trial: optuna.Trial, name: str)#

OptimizableField(

default: Any,

*,

space: SearchSpace | None = None,

merge_conflict: str = 'overwrite',

**fld_kw,

)#

class OptimizableMixin(/, **data: Any)#

Bases: pydantic.BaseModel

optimizable_params: list[str] = None#

search_space: dict[str, SearchSpace] = None#