1import numpy as np
4def create_gaussian(*, n_samples: int, stddev: float, seed: int) -> np.ndarray:
5 rng = np.random.default_rng(seed)
6 x = np.round(rng.normal(0, stddev, n_samples)).astype(np.int16)
7 return x
9datasets = [
10 {
11 'name': 'gaussian-1',
12 'version': '1',
13 'create': lambda: create_gaussian(n_samples=1_000_000, stddev=1, seed=0)
14 },
15 {
16 'name': 'gaussian-2',
17 'version': '1',
18 'create': lambda: create_gaussian(n_samples=1_000_000, stddev=2, seed=0)
19 },
20 {
21 'name': 'gaussian-3',
22 'version': '1',
23 'create': lambda: create_gaussian(n_samples=1_000_000, stddev=3, seed=0)
24 },
25 {
26 'name': 'gaussian-5',
27 'version': '1',
28 'create': lambda: create_gaussian(n_samples=1_000_000, stddev=5, seed=0)
29 },
30 {
31 'name': 'gaussian-8',
32 'version': '1',
33 'create': lambda: create_gaussian(n_samples=1_000_000, stddev=8, seed=0)
34 }
35]