Handle NaN coefficients

Eden-Kramer-Lab · Mar 19, 2018 · b4214ac · b4214ac
1 parent 8dc24b7
commit b4214ac
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/replay_classification/sorted_spikes.py b/replay_classification/sorted_spikes.py
@@ -49,7 +49,9 @@ def predictors_by_trajectory_direction(trajectory_direction,
 def get_conditional_intensity(fit_coefficients, predict_design_matrix):
     '''The conditional intensity for each model
     '''
-    return np.exp(np.dot(predict_design_matrix, fit_coefficients)).T
+    intensity = np.exp(np.dot(predict_design_matrix, fit_coefficients)).T
+    intensity[np.isnan(intensity)] = np.spacing(1)
+    return intensity
 
 
 def atleast_kd(array, k):