icecube.ml_suite.i3deepice module¶
All the things needed to replace i3deepice with ml_suite
- class icecube.ml_suite.i3deepice.I3DeepiceHexDataTransformer(config: str | dict, name: str, string_axis: int = 1, norm_transformations: List[Callable] | None = None)¶
Bases:
HexDataTransformer
Creates a data transformer for converting an input tensor’s string axis to a 2 axes hexagonal grid and applies normalization.
- Parameters:
config (str or dict) – Path to yaml file or a dictionary containing string_hex_dict transformation definition.
name (str) – Transformation name key in the config dictionary.
string_axis (int, optional) – The axis which defines the DOM string number in the input tensor x_input.
norm_transformations (list of callables, optional) – The features normalization list containing callables for each feature.
- Returns:
The output tensor which is now transformed to an hexagonal grid. Shape: […, 10, 10, …] with 10, 10 inserted at the specified string_axis.
- Return type:
array_like
- icecube.ml_suite.i3deepice.get_i3deepice_trafo(config)¶