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)