AMD EXA UTSStat

christianbrodbeck · Jun 16, 2021 · 8ebe161 · 8ebe161
1 parent df0a708
commit 8ebe161
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/eelbrain/datasets/_sim_eeg.py b/eelbrain/datasets/_sim_eeg.py
@@ -23,7 +23,7 @@ def _topo(sensor, center, falloff=1):
 def simulate_erp(
         n_trials: int = 80,
         seed: int = 0,
-        snr: float = 0.2,
+        snr: float = None,
 ) -> Dataset:
     """Simulate event-related EEG data
 
@@ -101,7 +101,10 @@ def simulate_erp(
     # Add noise
     noise = powerlaw_noise(signal, 1, rng)
     noise = noise.smooth('sensor', 0.02, 'gaussian')
-    noise *= (signal.std() / noise.std() / snr)
+    if snr is None:
+        noise *= 5
+    else:
+        noise *= (signal.std() / noise.std() / snr)
     signal += noise
 
     # Data scale