Source code for master project on flow matching for trans-dimensional simulation-based inference on fast astronomical transients.
To get started, clone the repository and install the required packages:
git clone https://github.com/Ninavd/FRB-flow.git
cd FRB-flow
pip install -r requirements.txt└── MCMC_runs/ # MCMC run results are stored here
| └── ...
├── notebooks/
│ ├── FRB_prep/ # Notebooks for processing raw FRB dynamic spectra
│ │ └── ...
│ ├── ... # Notebooks for evaluating trained FM models
├── scripts/
| ├── FM_training.py # training an FM model via the CLI
| └── MCMC.py # running MCMC sampling via the CLI
├── observational_data/
| ├── FRBs/ # pre-processed FRB profiles
| └── magnetars/ # raw magnetar bursts time series
├── src/
| └── flow_matching/
| ├── distributions.py
| ├── helpers.py
| ├── integration.py
| ├── loader.py
| ├── models.py
| ├── plotting.py
| ├── probability_path.py
| ├── simulator.py
| ├── training.py
| └── transformer.py
| └── MCMC/
| ├── helpers.py
│ ├── plotting.py
│ ├── posterior.py
│ └── priors.py
├── c2st.py