Datagen bns changes - Githubissues

ML4GW / aframe

Detecting binary black hole mergers in LIGO with neural networks

MIT License

18 stars 17 forks source link

Datagen bns changes #466

Open rafia17 opened 9 months ago

rafia17 commented 9 months ago

Following changes are implemented:

scripts/waveforms.py:

a signal_type variable is added. Based on its value, we will call generate_gw() for BBH and generate_gw_bns for BNS.
the call for generate_gw_bns is made through concurrent.futures.ProcessPoolExecutor to speed up waveform generation
some logging info is added to prompt when waveform generation is finished etc

utils/injection.py:

generate_gw_bns() is added. Its similar to generate_gw() except that it gets rid of the wraparound of the coalescence to the front of the signal.

projects/sandbox/pyproject.toml:

added signal_type variable with default set to "bbh"