1import numpy as np
2import os
5SOURCE_FILE = "bernoulli/__init__.py"
8def _load_long_description():
9 current_dir = os.path.dirname(os.path.abspath(__file__))
10 md_path = os.path.join(current_dir, "bernoulli.md")
11 with open(md_path, "r", encoding="utf-8") as f:
12 return f.read()
15LONG_DESCRIPTION = _load_long_description()
18def create_bernoulli(*, n_samples: int, p: float, seed: int) -> np.ndarray:
19 rng = np.random.default_rng(seed)
20 x = rng.binomial(1, p, n_samples).astype(np.uint8)
21 return x
24tags = ["bernoulli", "timeseries", "1d", "integer", "discrete", "synthetic", "i.i.d."]
26datasets = [
27 {
28 "name": "bernoulli-0.1",
29 "version": "3",
30 "create": lambda: create_bernoulli(n_samples=1_000_000, p=0.1, seed=0),
31 "description": "Binary sequence with 10% probability of ones.",
32 "tags": tags,
33 "source_file": SOURCE_FILE,
34 "long_description": LONG_DESCRIPTION,
35 },
36 {
37 "name": "bernoulli-0.2",
38 "version": "3",
39 "create": lambda: create_bernoulli(n_samples=1_000_000, p=0.2, seed=0),
40 "description": "Binary sequence with 20% probability of ones.",
41 "tags": tags,
42 "source_file": SOURCE_FILE,
43 "long_description": LONG_DESCRIPTION,
44 },
45 {
46 "name": "bernoulli-0.3",
47 "version": "3",
48 "create": lambda: create_bernoulli(n_samples=1_000_000, p=0.3, seed=0),
49 "description": "Binary sequence with 30% probability of ones.",
50 "tags": tags,
51 "source_file": SOURCE_FILE,
52 "long_description": LONG_DESCRIPTION,
53 },
54 {
55 "name": "bernoulli-0.4",
56 "version": "3",
57 "create": lambda: create_bernoulli(n_samples=1_000_000, p=0.4, seed=0),
58 "description": "Binary sequence with 40% probability of ones.",
59 "tags": tags,
60 "source_file": SOURCE_FILE,
61 "long_description": LONG_DESCRIPTION,
62 },
63 {
64 "name": "bernoulli-0.5",
65 "version": "3",
66 "create": lambda: create_bernoulli(n_samples=1_000_000, p=0.5, seed=0),
67 "description": "Binary sequence with 50% probability of ones and 50% probability of zeros.",
68 "tags": tags,
69 "source_file": SOURCE_FILE,
70 "long_description": LONG_DESCRIPTION,
71 },
72]