diff --git a/mu_map/polar_map/eval_pixelwise.py b/mu_map/polar_map/eval_pixelwise.py index 1d9d9d01d1decd4d8a4a9eb0c4942bbf0017b38c..e65837c191c9d9e3df8715005364c2c4011b8ee3 100644 --- a/mu_map/polar_map/eval_pixelwise.py +++ b/mu_map/polar_map/eval_pixelwise.py @@ -6,6 +6,7 @@ import matplotlib as mpl import matplotlib.pyplot as plt import numpy as np import pandas as pd +from scipy.stats import pearsonr from mu_map.polar_map.prepare import headers @@ -97,6 +98,7 @@ bins_y = np.arange(0, 100.1, 2) fig, axs = plt.subplots(2, 3, figsize=(15, 10)) for i, (label, recon) in enumerate(recons_tot.items()): recon = np.concatenate(recon) + print(f"{label}: {pearsonr(baseline_tot, recon)}") h, xedges, yedges, image = axs[0, i].hist2d(baseline_tot, recon, bins=(bins_x, bins_y)) h = (h - h.min()) / (h.max() - h.min())