supaernova.steps.data

source module supaernova.steps.data

Classes

• DataStep

source class DataStep(config: DataStepConfig)

Bases : SNPAEStep

Methods

• load_sne

• calculate_salt_flux

• get_dims

• prepare_data_arrays

• split_train_test

source method DataStep.load_sne() → None

source method DataStep.calculate_salt_flux() → None

source method DataStep.get_dims() → None

source method DataStep.prepare_data_arrays() → None

source method DataStep.split_train_test() → None