From c6daae5ff673384d9195c49b64fc260608c12235 Mon Sep 17 00:00:00 2001 From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de> Date: Thu, 15 Dec 2022 10:11:29 +0100 Subject: [PATCH] include correlation coefficient in pixelwise evaluation --- mu_map/polar_map/eval_pixelwise.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mu_map/polar_map/eval_pixelwise.py b/mu_map/polar_map/eval_pixelwise.py index 1d9d9d0..e65837c 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()) -- GitLab