model.detection package

Submodules

model.detection.dataloader module

class model.detection.dataloader.GWSEDataset

Bases: object

load_waveform(DIR='.', data_fn='waveform_dataset.hdf5')

Load the waveform dataset from a hdf5 file

Parameters:

• DIR (str, optional) – Directory of the data. Defaults to “.”.

• data_fn (str, optional) – Filename of the data. Defaults to “waveform_dataset.hdf5”.

save_waveform(DIR='.', data_fn='waveform_dataset.hdf5')

Save the waveform dataset to a hdf5 file

Parameters:

• DIR (str, optional) – Directory to save the data. Defaults to “.”.

• data_fn (str, optional) – Filename of the data. Defaults to “waveform_dataset.hdf5”.

class model.detection.dataloader.WaveformDatasetTorch(*args: Any, **kwargs: Any)

Bases: Dataset

merge_dataset()

Merge the waveform dataset and noise dataset

Module contents

Detect EMRI signal using DNN

model.detection.get_version() → str