From 0a8cafd000cd3a223bcb63adfa982744ac3d6d5d Mon Sep 17 00:00:00 2001
From: Danilo Ferreira de Lima <danilo.enoque.ferreira.de.lima@xfel.de>
Date: Tue, 11 Apr 2023 13:34:46 +0200
Subject: [PATCH] Adds non-linear terms between the pulse energy and
low-resolution data.
---
pes_to_spec/model.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/pes_to_spec/model.py b/pes_to_spec/model.py
index ce57af8..e8fe9ef 100644
--- a/pes_to_spec/model.py
+++ b/pes_to_spec/model.py
@@ -504,10 +504,10 @@ class SelectRelevantLowResolution(TransformerMixin, BaseEstimator):
for channel, item in X.items()}
if not keep_dictionary_structure:
selected = list(y.values())
- if self.poly:
- selected += [np.sqrt(np.fabs(v)) for v in y.values()]
if pulse_energy is not None:
selected += [pulse_energy[:, np.newaxis, :]]
+ if self.poly:
+ selected += [pulse_energy[:, np.newaxis, :]*v for v in y.values()]
return np.concatenate(selected, axis=-1)
return y
--
GitLab