From fe32298ddf9aa30b5a33f054876eda36c54405eb Mon Sep 17 00:00:00 2001
From: Danilo Ferreira de Lima <danilo.enoque.ferreira.de.lima@xfel.de>
Date: Mon, 16 Jan 2023 16:29:53 +0100
Subject: [PATCH] Added method to return x-axis from high resolution data.
---
pes_to_spec/model.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/pes_to_spec/model.py b/pes_to_spec/model.py
index d86499c..0e253f2 100644
--- a/pes_to_spec/model.py
+++ b/pes_to_spec/model.py
@@ -66,7 +66,7 @@ class HighResolutionSmoother(TransformerMixin, BaseEstimator):
Returns: The object itself.
"""
- self.energy = fit_params["energy"]
+ self.energy = np.copy(fit_params["energy"])
if len(self.energy.shape) == 2:
self.energy = self.energy[0,:]
return self
@@ -536,6 +536,10 @@ class Model(TransformerMixin, BaseEstimator):
"""Get channels used in training."""
return self.x_model.named_steps["select"].channels
+ def get_energy_values(self) -> np.ndarray:
+ """Get x-axis of high-resolution data."""
+ return self.y_model.named_steps["smoothen"].energy
+
def debug_peak_finding(self, low_res_data: Dict[str, np.ndarray], filename: str):
"""
Produce image to understand if the peak finding step worked well.
--
GitLab