Symmetry¶

class baybe.symmetries.Symmetry[source]¶

Bases: SerialMixin, ABC

Abstract base class for symmetries.

Symmetry is a concept that can be used to configure the modelling process in the presence of invariances.

Public methods

`__init__`(*[, use_data_augmentation])	Method generated by attrs for class Symmetry.
`augment_measurements`(df, parameters)	Augment the given measurements according to the symmetry.
`from_dict`(dictionary)	Create an object from its dictionary representation.
`from_json`(source, /)	Create an object from its JSON representation.
`summary`()	Return a custom summarization of the symmetry.
`to_dict`()	Create an object's dictionary representation.
`to_json`([sink, overwrite])	Create an object's JSON representation.
`validate_searchspace_context`(searchspace)	Validate that the symmetry is compatible with the given searchspace.

Public attributes and properties

`use_data_augmentation`	Flag indicating whether data augmentation would be used with surrogates that support this.
`parameter_names`	The names of the parameters affected by the symmetry.

__init__(*, use_data_augmentation: bool = True)¶

Method generated by attrs for class Symmetry.

For details on the parameters, see Public attributes and properties.

abstract augment_measurements(df: pd.DataFrame, parameters: Iterable[Parameter])[source]¶

Augment the given measurements according to the symmetry.

Parameters:

df (pd.DataFrame) – The dataframe containing the measurements to be augmented.
parameters (Iterable[Parameter]) – Parameter objects carrying additional information (might not be needed by all augmentation implementations).

Return type:

pd.DataFrame

Returns:

The augmented dataframe including the original measurements.

classmethod from_dict(dictionary: dict)¶

Create an object from its dictionary representation.

Parameters:: dictionary (dict) – The dictionary representation.
Return type:: TypeVar(_T, bound= SerialMixin)
Returns:: The reconstructed object.

classmethod from_json(source: str | Path | SupportsRead[str], /)¶

Create an object from its JSON representation.

Parameters:

source (str | Path | SupportsRead[str]) –

The JSON source. Can be:

A string containing JSON content.
A file path or Path object pointing to a JSON file.
A file-like object with a read() method.

Raises:

ValueError – If source is not one of the allowed types.

Return type:

Returns:

The reconstructed object.

summary()[source]¶

Return a custom summarization of the symmetry.

Return type:: dict

to_dict()¶

Create an object’s dictionary representation.

Return type:: dict
Returns:: The dictionary representation of the object.

to_json(sink: str | Path | SupportsWrite[str] | None = None, /, *, overwrite: bool = False, **kwargs: Any)¶

Create an object’s JSON representation.

Parameters:

sink (str | Path | SupportsWrite[str] | None) –
The JSON sink. Can be:
- None (only returns the JSON string).
- A file path or Path object pointing to a location where to write the JSON content.
- A file-like object with a write() method.
overwrite (bool) – Boolean flag indicating if to overwrite the file if it already exists. Only relevant if sink is a file path or Path object.
**kwargs (Any) – Additional keyword arguments to pass to json.dumps().

Raises:

FileExistsError – If sink points to an already existing file but overwrite is False.

Return type:

str

Returns:

The JSON representation as a string.

validate_searchspace_context(searchspace: SearchSpace)[source]¶

Validate that the symmetry is compatible with the given searchspace.

Parameters:: searchspace (SearchSpace) – The searchspace to validate against.
Raises:: ValueError – If the symmetry affects parameters not present in the searchspace.
Return type:: None

abstract property parameter_names: tuple[str, ...]¶: The names of the parameters affected by the symmetry.

use_data_augmentation: bool¶: Flag indicating whether data augmentation would be used with surrogates that support this.