Use n_background for background events

cta-observatory · Nov 14, 2023 · c70d36b · c70d36b
1 parent c444700
commit c70d36b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ctapipe/tools/train_particle_classifier.py b/ctapipe/tools/train_particle_classifier.py
@@ -193,7 +193,7 @@ def _read_input_data(self, tel_type):
             feature_names=feature_names,
             rng=self.rng,
             log=self.log,
-            n_events=self.n_signal.tel[tel_type],
+            n_events=self.n_background.tel[tel_type],
         )
         table = vstack([signal, background])
         self.log.info(