fix photon computation with feature scaling

NNPDF · Oct 4, 2024 · 04c45aa · 04c45aa
1 parent 01b26a0
commit 04c45aa
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/n3fit/src/n3fit/model_trainer.py b/n3fit/src/n3fit/model_trainer.py
@@ -936,7 +936,10 @@ def hyperparametrizable(self, params):
             )
 
             if photons:
-                pdf_model.get_layer("add_photon").register_photon(xinput.input.tensor_content)
+                if self._scaler: # select only the non-scaled input
+                    pdf_model.get_layer("add_photon").register_photon(xinput.input.tensor_content[:,:,1:])
+                else:
+                    pdf_model.get_layer("add_photon").register_photon(xinput.input.tensor_content)
 
             # Model generation joins all the different observable layers
             # together with pdf model generated above