1"""Custom physical-support track: parent activity from noisy nearly-colliding rays."""
 2import numpy as np
 3
 4META = {
 5    "name": "collision_support",
 6    "domain": "embedding",
 7    "description": "Parent-block activity from replicated noisy observations of two nearly colliding sign-invariant child rays.",
 8}
 9
10
11def get_dataset(seed, n_train, n_test):
12    rng = np.random.default_rng(seed)
13    s = 0.20
14    sigma = 0.35
15
16    def make(n):
17        active = rng.integers(0, 2, size=n).astype(np.int64)
18        x = rng.normal(0, sigma, size=(n, 4, 2)).astype(np.float32)
19        a = np.arcsin(s / 2.0)
20        rays = np.array([[np.cos(a), np.sin(a)], [np.cos(a), -np.sin(a)]], dtype=np.float32)
21        for i in range(n):
22            if active[i]:
23                for t in range(4):
24                    k = rng.integers(0, 2)
25                    sign = rng.choice([-1.0, 1.0])
26                    x[i, t] += sign * rays[k]
27        return x.reshape(n, -1), active
28
29    xtr, ytr = make(n_train)
30    xte, yte = make(n_test)
31    return {
32        "xtr": xtr, "ytr": ytr, "xte": xte, "yte": yte,
33        "task": "classification", "metric": "err",
34        "input_shape": (8,), "out_dim": 2,
35        "s": s, "sigma": sigma, "N": n_train, "T": 4,
36    }