Minor update

IdeasLabUT · Apr 30, 2021 · 16138db · 16138db
1 parent f3d3d2e
commit 16138db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/generative_model_utils.py b/generative_model_utils.py
@@ -79,8 +79,8 @@ def simulate_univariate_hawkes(mu, alpha, beta, run_time, seed=None):
         np.random.seed(seed)
 
     # Creating a numpy vector with 3 standard deviations from the expected number of events (99.73% of cases)
-    expected_num_events = max((mu * run_time) / (1 - (alpha / beta)), 5)
-    sd = max(int(np.sqrt((mu * run_time) / ((1 - (alpha / beta)) ** 3))), 5)
+    expected_num_events = max((mu * run_time) / (1 - (alpha / beta)), 4)
+    sd = max(int(np.sqrt((mu * run_time) / ((1 - (alpha / beta)) ** 3))), 4)
     num_events_upper_limit = int(expected_num_events + 3 * sd)
 
     timestamps = np.zeros(num_events_upper_limit, dtype=np.float32)