Update likelihood bounds in GW150914_PV2.py

kazewong · Jan 30, 2024 · 309a22d · 309a22d
1 parent 9ed508f
commit 309a22d
Showing 1 changed file with 22 additions and 2 deletions.
diff --git 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)