/ proof-of-concept / benchcompress
Sign in
proof-of-concept / benchcompress
69b432b
Branches
mainbenchmark-results
benchcompress / zia_benchmark / src / zia_benchmark / datasets / gaussian / __init__.py
56 lines · 1.6 KBCodeBlameHistory
e892a17zia_benchmarkJeremy Magland 1import numpy as np
db4e6b9links to source codeJeremy Magland 4SOURCE_FILE = "gaussian/__init__.py"
e892a17zia_benchmarkJeremy Magland 7def create_gaussian(*, n_samples: int, stddev: float, seed: int) -> np.ndarray:
8 rng = np.random.default_rng(seed)
9 x = np.round(rng.normal(0, stddev, n_samples)).astype(np.int16)
10 return x
70a7a71code formattingJeremy Magland 12
769ad9aformatJeremy Magland 13tags = ["integer", "discrete", "synthetic"]
e892a17zia_benchmarkJeremy Magland 15datasets = [
16 {
70a7a71code formattingJeremy Magland 17 "name": "gaussian-1",
18 "version": "1",
19 "create": lambda: create_gaussian(n_samples=1_000_000, stddev=1, seed=0),
85d3449pagesJeremy Magland 20 "description": "Rounded Gaussian integers with σ=1.",
769ad9aformatJeremy Magland 21 "tags": tags,
db4e6b9links to source codeJeremy Magland 22 "source_file": SOURCE_FILE,
e892a17zia_benchmarkJeremy Magland 23 },
24 {
70a7a71code formattingJeremy Magland 25 "name": "gaussian-2",
26 "version": "1",
27 "create": lambda: create_gaussian(n_samples=1_000_000, stddev=2, seed=0),
85d3449pagesJeremy Magland 28 "description": "Rounded Gaussian integers with σ=2.",
769ad9aformatJeremy Magland 29 "tags": tags,
db4e6b9links to source codeJeremy Magland 30 "source_file": SOURCE_FILE,
e892a17zia_benchmarkJeremy Magland 31 },
32 {
70a7a71code formattingJeremy Magland 33 "name": "gaussian-3",
34 "version": "1",
35 "create": lambda: create_gaussian(n_samples=1_000_000, stddev=3, seed=0),
85d3449pagesJeremy Magland 36 "description": "Rounded Gaussian integers with σ=3.",
769ad9aformatJeremy Magland 37 "tags": tags,
db4e6b9links to source codeJeremy Magland 38 "source_file": SOURCE_FILE,
e892a17zia_benchmarkJeremy Magland 39 },
40 {
70a7a71code formattingJeremy Magland 41 "name": "gaussian-5",
42 "version": "1",
43 "create": lambda: create_gaussian(n_samples=1_000_000, stddev=5, seed=0),
85d3449pagesJeremy Magland 44 "description": "Rounded Gaussian integers with σ=5.",
769ad9aformatJeremy Magland 45 "tags": tags,
db4e6b9links to source codeJeremy Magland 46 "source_file": SOURCE_FILE,
e892a17zia_benchmarkJeremy Magland 47 },
48 {
70a7a71code formattingJeremy Magland 49 "name": "gaussian-8",
50 "version": "1",
51 "create": lambda: create_gaussian(n_samples=1_000_000, stddev=8, seed=0),
85d3449pagesJeremy Magland 52 "description": "Rounded Gaussian integers with σ=8.",
769ad9aformatJeremy Magland 53 "tags": tags,
db4e6b9links to source codeJeremy Magland 54 "source_file": SOURCE_FILE,
70a7a71code formattingJeremy Magland 55 },
e892a17zia_benchmarkJeremy Magland 56]
↑↓ move↵openescclose