From 44b0c011fbd9e38c3246830bae7ffee784d57c5a Mon Sep 17 00:00:00 2001 From: Christian Chapman-Bird Date: Mon, 27 Sep 2021 15:01:00 +0100 Subject: [PATCH] heatmap function fix --- EMRI_DET/validate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/EMRI_DET/validate.py b/EMRI_DET/validate.py index 190a63f..a6476ef 100644 --- a/EMRI_DET/validate.py +++ b/EMRI_DET/validate.py @@ -216,7 +216,7 @@ def grid_heatmap_corner(dataframe, truth_column, pred_column, log=True, ratio=Fa vmin = np.min([np.min(hmap) for hmap in plotmaps]) vmax = np.max([np.max(hmap) for hmap in plotmaps]) - fig, ax = plt.subplots(ncols=5, nrows=5, figsize=(8 + int(2*nparams),6 + int(2*nparams))) + fig, ax = plt.subplots(ncols=nparams, nrows=nparams, figsize=(8 + int(2*nparams),6 + int(2*nparams))) plt.subplots_adjust(wspace=0.05, hspace=0.05) num = 0 for i in range(nparams): -- GitLab