├── .DS_Store ├── AUTHORS.md ├── KENN-logo-bg.png ├── LICENSE ├── README.md ├── clustering.py ├── config ├── __init__.py ├── kenn_config.yml └── population_priors.yml ├── dataset ├── __init__.py ├── torch_dataset.py └── whitening.py ├── model ├── __init__.py ├── flow_utils.py ├── kenn.py ├── kenn_flow.py └── kenn_flow_builder.py ├── pp_plot.py ├── run_train.py ├── simulations ├── __init__.py ├── einstein_telescope.py ├── utils.py └── waveform_generator.py ├── test_train.py └── training ├── __init__.py └── trainer.py /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/.DS_Store -------------------------------------------------------------------------------- /AUTHORS.md: -------------------------------------------------------------------------------- 1 | Code authors: Lucia Papalini and Federico De Santi 2 | -------------------------------------------------------------------------------- /KENN-logo-bg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/KENN-logo-bg.png -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/README.md -------------------------------------------------------------------------------- /clustering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/clustering.py -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/kenn_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/config/kenn_config.yml -------------------------------------------------------------------------------- /config/population_priors.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/config/population_priors.yml -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/torch_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/dataset/torch_dataset.py -------------------------------------------------------------------------------- /dataset/whitening.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/dataset/whitening.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | from .kenn import * -------------------------------------------------------------------------------- /model/flow_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/model/flow_utils.py -------------------------------------------------------------------------------- /model/kenn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/model/kenn.py -------------------------------------------------------------------------------- /model/kenn_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/model/kenn_flow.py -------------------------------------------------------------------------------- /model/kenn_flow_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/model/kenn_flow_builder.py -------------------------------------------------------------------------------- /pp_plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/pp_plot.py -------------------------------------------------------------------------------- /run_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/run_train.py -------------------------------------------------------------------------------- /simulations/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/simulations/__init__.py -------------------------------------------------------------------------------- /simulations/einstein_telescope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/simulations/einstein_telescope.py -------------------------------------------------------------------------------- /simulations/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/simulations/utils.py -------------------------------------------------------------------------------- /simulations/waveform_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/simulations/waveform_generator.py -------------------------------------------------------------------------------- /test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/test_train.py -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/training/__init__.py -------------------------------------------------------------------------------- /training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luciapapalini/kenn-gw-transformer/HEAD/training/trainer.py --------------------------------------------------------------------------------