1import numpy as np
 2META = {"name":"relativistic_conservative_regression","domain":"conservative_state_admissibility","description":"Predict valid relativistic conservative states from valid baselines and controls; GQL residual limiting is structurally native."}
 3
 4def _make(seed, n):
 5    r=np.random.RandomState(seed)
 6    # Inputs are a valid baseline U0=(D,mx,my,E0) plus a requested physical perturbation.
 7    d=r.uniform(.4,2.0,n); mx=r.normal(0,.35,n); my=r.normal(0,.35,n)
 8    e=np.sqrt(d*d+mx*mx+my*my)+r.uniform(.15,.8,n)
 9    u0=np.stack([d,mx,my,e],1)
10    # A smooth target state; training noise makes unconstrained extrapolations common.
11    dd=.45*np.tanh(r.normal(size=n)); dm=r.normal(0,.38,(n,2));
12    target=u0.copy(); target[:,0]=d+dd
13    target[:,1:3]+=dm
14    target[:,3]=np.sqrt(target[:,0]**2+np.sum(target[:,1:3]**2,1))+r.uniform(.08,.65,n)
15    # The model receives baseline and a control-like requested delta.
16    x=np.concatenate([u0, target-u0],1).astype(np.float32)
17    return x,target.astype(np.float32)
18
19def get_dataset(seed,n_train,n_test):
20    xtr,ytr=_make(seed,n_train); xte,yte=_make(seed+5000,n_test)
21    return {"xtr":xtr,"ytr":ytr,"xte":xte,"yte":yte,"task":"regression","metric":"mse","out_dim":4}