modified performance analyzer to handle case where shot is shorter than T_min_warn

Julian Kates-Harbeck · Julian Kates-Harbeck · commit 34f66b67549a · 2018-03-19T15:52:09.000-04:00
diff --git a/plasma/utils/performance.py b/plasma/utils/performance.py
@@ -188,8 +188,11 @@ def get_threshold_arrays(self,preds,truths,disruptives):
                         d_early_thresholds.append(-np.inf)
                 else:
                         d_early_thresholds.append(np.max(pred[early_indices]))
-                        
-                d_correct_thresholds.append(np.max(pred[correct_indices]))
+
+                if np.sum(correct_indices) == 0:
+                    d_correct_thresholds.append(-np.inf)
+                else:
+                    d_correct_thresholds.append(np.max(pred[correct_indices]))
             else:
                 nd_thresholds.append(np.max(pred))
         return np.array(d_early_thresholds), np.array(d_correct_thresholds),np.array(d_late_thresholds), np.array(nd_thresholds)
