The inference demo are in inference.ipynb.
Please include the following references in your articles if you use resources from this repository:
@article{Tian:2023vot, author = "Tian, Minyang and Huerta, E. A. and Zheng, Huihuo", title = "{Physics-inspired spatiotemporal-graph AI ensemble for gravitational wave detection}", eprint = "2306.15728", archivePrefix = "arXiv", primaryClass = "astro-ph.IM", month = "6", year = "2023" }
and
@article{Tian:2023epx, author = "Tian, Minyang and Huerta, E. A. and Zheng, Huihuo", title = "{AI ensemble for signal detection of higher order gravitational wave modes of quasi-circular, spinning, non-precessing binary black hole mergers}", eprint = "2310.00052", archivePrefix = "arXiv", primaryClass = "astro-ph.IM", month = "9", year = "2023" }