Published April 2, 2026 | Version v1
Dataset Open

Trained models for gravitational wave inference with `labrador`

  • 1. ROR icon California Institute of Technology

Description

Trained labrador models

This record contains trained models to do amortized gravitational wave inference with labrador, and a jupyter notebook to illustrate their usage.

For demonstration purposes, it also contains data files corresponding to events in GWTC-4.0. These files have been created with cogwheel; refer to its documentation for how to create them.

Note: since these data files are only intended for demonstration, they all have a frequency range of (15, 1024) Hz (even though LVK may have flagged some frequency ranges unusable for some events, or the some signals may have power at higher frequencies), and all are 64s long with the event at the center. Minimal processing has been applied in the form of inpainting loud glitches.

Installation

A conda environment specification file environment.yml has been provided for reproducibility. Create the environment with

conda env create -f environment.yml
conda activate labrador-demo

and run the notebook inference.ipynb within it.

Versions

  •  The models in v1 of the zenodo record were created with labrador's commit c23b81476139e760195fcc266e087fe7f042f1da

Reference

Javier Roulet, Marco Crisostomi, Lucy M Thomas, Katerina Chatziioannou (2026). labrador: a domain-optimized machine-learning tool for gravitational wave inference [2604.08897]

Files

GWTC-4.0_event_data.zip

Files (1.1 GB)

Name Size Download all
md5:e6d6cfc6e5036ef7bebd5518199d26cd
1.0 kB Download
md5:72adf7ef904bcb43b6e4a3da995c5bf9
373.5 MB Preview Download
md5:e26a740b10b92ba733a9067847127dd8
2.4 MB Preview Download
md5:2b0a205adeff61174fd765cac9134ea6
731.1 MB Preview Download
md5:1d86a47eacd06fd45c9286d6b4c658a8
1.2 kB Preview Download