Energy-Riesz checkpoint selector / poisson_track.py
Mechanism confirmed, baseline not beaten
1import numpy as np
2META = {"name":"poisson_manufactured", "domain":"pde", "description":"2D Dirichlet Poisson manufactured-solution regression track"}
3def get_dataset(seed, n_train, n_test):
4 rng=np.random.default_rng(seed)
5 def pts(n): return rng.uniform(0.015,0.985,size=(n,2)).astype(np.float32)
6 def sol(x): return (np.sin(np.pi*x[:,0])*np.sin(np.pi*x[:,1])).astype(np.float32)[:,None]
7 xtr,xte=pts(n_train),pts(n_test)
8 return {"xtr":xtr,"ytr":sol(xtr),"xte":xte,"yte":sol(xte),"task":"regression","metric":"mse","out_dim":1}