Count invalid events as non-predictable, not valid events

cta-observatory · Nov 7, 2023 · 9815e7a · 9815e7a
1 parent 799d937
commit 9815e7a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/ctapipe/tools/train_disp_reconstructor.py b/ctapipe/tools/train_disp_reconstructor.py
@@ -168,7 +168,7 @@ def _read_table(self, telescope_type):
 
             valid = check_valid_rows(table_chunk)
             if not np.all(valid):
-                n_non_predictable += np.sum(valid)
+                n_non_predictable += np.sum(~valid)
                 table_chunk = table_chunk[valid]
 
             table.append(table_chunk)

diff --git a/ctapipe/tools/train_energy_regressor.py b/ctapipe/tools/train_energy_regressor.py
@@ -151,7 +151,7 @@ def _read_table(self, telescope_type):
 
             valid = check_valid_rows(table_chunk)
             if not np.all(valid):
-                n_non_predictable += np.sum(valid)
+                n_non_predictable += np.sum(~valid)
                 table_chunk = table_chunk[valid]
 
             table.append(table_chunk)

diff --git a/ctapipe/tools/train_particle_classifier.py b/ctapipe/tools/train_particle_classifier.py
@@ -202,7 +202,7 @@ def _read_table(self, telescope_type, loader, n_events=None):
 
             valid = check_valid_rows(table_chunk)
             if not np.all(valid):
-                n_non_predictable += np.sum(valid)
+                n_non_predictable += np.sum(~valid)
                 table_chunk = table_chunk[valid]
 
             table.append(table_chunk)