icecube.ml_suite.hdf module

icecube.ml_suite.hdf.read_input_features(hdf_file, n_feature=None, ml_feature_key='ml_suite_features')

Read Input Features from HDF file

Parameters:
  • hdf_file (str) – The path to the HDF file from which to read the input features.

  • n_feature (int, optional) – The number of input features per DOM. If None, this number is inferred from the hdf file.

  • ml_feature_key (str, optional) – The name of the frame key to which the features were written to.

Returns:

The input features. Shape: [n_events, 86, 60, n_feature]

Return type:

array_like