From 309a22d3da1b01ee68fd0079005f670a759eac77 Mon Sep 17 00:00:00 2001 From: Kaze Wong Date: Tue, 30 Jan 2024 15:52:38 -0500 Subject: [PATCH] Update likelihood bounds in GW150914_PV2.py --- example/GW150914_PV2.py | 24 ++++++++++++++++++++++-- 1 file changed, 22 insertions(+), 2 deletions(-) diff --git a/example/GW150914_PV2.py b/example/GW150914_PV2.py index 134c45bf..a36875e5 100644 --- a/example/GW150914_PV2.py +++ b/example/GW150914_PV2.py @@ -89,8 +89,28 @@ sin_dec_prior, ], ) -likelihood = TransientLikelihoodFD([H1, L1], waveform=waveform, trigger_time=gps, duration=4, post_trigger_duration=2) -# likelihood = HeterodynedTransientLikelihoodFD([H1, L1], prior=prior, bounds=[prior.xmin, prior.xmax], waveform=RippleIMRPhenomD(), trigger_time=gps, duration=4, post_trigger_duration=2) + +bounds = jnp.array( + [ + [10.0, 80.0], + [0.125, 1.0], + [0, jnp.pi], + [0, 2*jnp.pi], + [0.0, 1.0], + [0, jnp.pi], + [0, 2*jnp.pi], + [0.0, 1.0], + [0.0, 2000.0], + [-0.05, 0.05], + [0.0, 2 * jnp.pi], + [-1.0, 1.0], + [0.0, jnp.pi], + [0.0, 2 * jnp.pi], + [-1.0, 1.0], + ] +) +# likelihood = TransientLikelihoodFD([H1, L1], waveform=waveform, trigger_time=gps, duration=4, post_trigger_duration=2) +likelihood = HeterodynedTransientLikelihoodFD([H1, L1], prior=prior, bounds=bounds, waveform=waveform, trigger_time=gps, duration=4, post_trigger_duration=2) mass_matrix = jnp.eye(prior.n_dim)