From 0c23a584952260dba5ff3edb93b335d55246e618 Mon Sep 17 00:00:00 2001 From: Danilo Ferreira de Lima <danilo.enoque.ferreira.de.lima@xfel.de> Date: Tue, 31 Jan 2023 18:00:37 +0100 Subject: [PATCH] Corrected compatibility check. --- pes_to_spec/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pes_to_spec/model.py b/pes_to_spec/model.py index 55bd17d..0ed420a 100644 --- a/pes_to_spec/model.py +++ b/pes_to_spec/model.py @@ -652,9 +652,9 @@ class Model(TransformerMixin, BaseEstimator): """ selection_model = self.x_model['select'] low_res = selection_model.transform(low_res_data, keep_dictionary_structure=True) - pca_model = self.channel_pca_model.named_steps['pca'] quality = {channel: 0.0 for channel in low_res.keys()} for channel in low_res.keys(): + pca_model = self.channel_pca_model[channel].named_steps['pca'] low_pca = pca_model.transform(low_res[channel]) low_pca_rec = pca_model.inverse_transform(low_pca) low_pca_unc = self.channel_pca_model.named_steps['unc'].uncertainty() -- GitLab