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