1import numpy as np
4SOURCE_FILE = "bernoulli/__init__.py"
7def create_bernoulli(*, n_samples: int, p: float, seed: int) -> np.ndarray:
8 rng = np.random.default_rng(seed)
9 x = rng.binomial(1, p, n_samples).astype(np.uint8)
10 return x
13datasets = [
14 {
15 "name": "bernoulli-0.1",
16 "version": "3",
17 "create": lambda: create_bernoulli(n_samples=1_000_000, p=0.1, seed=0),
18 "description": "Binary sequence with 10% probability of ones.",
19 "tags": ["binary"],
20 "source_file": SOURCE_FILE,
21 },
22 {
23 "name": "bernoulli-0.2",
24 "version": "3",
25 "create": lambda: create_bernoulli(n_samples=1_000_000, p=0.2, seed=0),
26 "description": "Binary sequence with 20% probability of ones.",
27 "tags": ["binary"],
28 "source_file": SOURCE_FILE,
29 },
30 {
31 "name": "bernoulli-0.3",
32 "version": "3",
33 "create": lambda: create_bernoulli(n_samples=1_000_000, p=0.3, seed=0),
34 "description": "Binary sequence with 30% probability of ones.",
35 "tags": ["binary"],
36 "source_file": SOURCE_FILE,
37 },
38 {
39 "name": "bernoulli-0.4",
40 "version": "3",
41 "create": lambda: create_bernoulli(n_samples=1_000_000, p=0.4, seed=0),
42 "description": "Binary sequence with 40% probability of ones.",
43 "tags": ["binary"],
44 "source_file": SOURCE_FILE,
45 },
46 {
47 "name": "bernoulli-0.5",
48 "version": "3",
49 "create": lambda: create_bernoulli(n_samples=1_000_000, p=0.5, seed=0),
50 "description": "Binary sequence with 50% probability of ones and 50% probability of zeros.",
51 "tags": ["binary"],
52 "source_file": SOURCE_FILE,
53 },
54]