From b8afe00ce317f06b8a25bf0a9088d0c89ead4c3d Mon Sep 17 00:00:00 2001
From: Philipp Arras <parras@mpa-garching.mpg.de>
Date: Tue, 16 Jul 2019 11:27:03 +0200
Subject: [PATCH] Fix plotting
---
helpers/plot.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/helpers/plot.py b/helpers/plot.py
index 4a742c3..1756edf 100644
--- a/helpers/plot.py
+++ b/helpers/plot.py
@@ -199,7 +199,7 @@ def plot_reconstruction_2d(data, ground_truth, KL, signal, R, A, name):
label='reconstruction')
ax[1, 2].plot(
amp_mean.domain[0].k_lengths,
- A.force(ground_truth).to_global_data(),
+ A.force(ground_truth).to_global_data()**2,
color='b',
label='ground truth')
ax[1, 2].legend()
--
GitLab