-
Notifications
You must be signed in to change notification settings - Fork 0
/
sim_adv_event.py
37 lines (30 loc) · 884 Bytes
/
sim_adv_event.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import numpy as np
import typer
from advanced_model import tau_series_generator as advanced_model
from sim_utils import simulate_event_stats
def main(
out_filename_template: str,
*,
sigma: float = 1e-4,
boundary: float = 1e-4,
hurst: float = 0.5,
dt: float = 1,
n_batches: int = 64,
n_batch_points: int = 1048576,
seed: int = -1,
):
beta_pdf, beta_psd = simulate_event_stats(
advanced_model,
sigma=sigma,
hurst=hurst,
boundary=boundary,
dt=dt,
n_batch_points=n_batch_points,
seed=seed,
)
out_filename = out_filename_template.format("pdf")
np.savetxt(out_filename, beta_pdf, fmt="%.4f", delimiter=",")
out_filename = out_filename_template.format("psd")
np.savetxt(out_filename, beta_psd, fmt="%.4f", delimiter=",")
if __name__ == "__main__":
typer.run(main)