Commit 44b0c011 authored by Christian Chapman-Bird's avatar Christian Chapman-Bird
Browse files

heatmap function fix

parent fc7310da

EMRI_DET/validate.py

+1 −1
Original line number Diff line number Diff line
@@ -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):