Merge pull request #133 from fact-project/fix_plotting_cta

Fix disp performance plot for cta
fact-project · May 27, 2020 · 5a5b1af · 5a5b1af
2 parents 037486a + 017ea92
commit 5a5b1af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aict_tools/plotting.py b/aict_tools/plotting.py
@@ -327,7 +327,7 @@ def plot_energy_dependent_disp_metrics(df, true_energy_column, energy_unit='GeV'
         np.log10(df[true_energy_column].max()),
         0.2
     )
-    df['bin_idx'] = np.digitize(df['corsika_event_header_total_energy'], edges)
+    df['bin_idx'] = np.digitize(df[true_energy_column], edges)
 
     def accuracy(group):
         return metrics.accuracy_score(