From 1055bb568000da36c8a5c89cc56a79132631edcc Mon Sep 17 00:00:00 2001 From: Ravi Kumar Date: Fri, 11 Oct 2024 00:36:28 +0530 Subject: [PATCH] save searched area array along with plots (#146) --- amplfi/train/models/flow.py | 1 + 1 file changed, 1 insertion(+) diff --git a/amplfi/train/models/flow.py b/amplfi/train/models/flow.py index 5ad5ee0b..6e85c22e 100644 --- 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)