More explicit place std deviation

Eden-Kramer-Lab · Nov 20, 2017 · 15f716c · 15f716c
1 parent ff52e95
commit 15f716c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/replay_classification/decoders.py b/replay_classification/decoders.py
@@ -101,7 +101,9 @@ def fit(self):
             self.position.min(), self.position.max(),
             self.n_position_bins + 1)
         if self.place_std_deviation is None:
-            self.place_std_deviation = 2 * np.diff(self.place_bin_edges)[0]
+            self.place_std_deviation = (
+                (self.position.max() - self.position.min()) /
+                self.n_position_bins)
         self.place_bin_centers = get_bin_centers(self.place_bin_edges)
 
         trajectory_directions = np.unique(