save searched area array along with plots (ML4GW#146)

EthanMarx · Oct 10, 2024 · 1055bb5 · 1055bb5
1 parent 77fe28d
commit 1055bb5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/amplfi/train/models/flow.py b/amplfi/train/models/flow.py
@@ -185,3 +185,4 @@ def on_test_epoch_end(self):
         plt.ylabel("Cumulative Probability")
         plt.title("Searched Area Cumulative Distribution Function")
         plt.savefig(self.outdir / "searched_area.png")
+        np.save(self.outdir / "searched_area.npy", searched_areas)