Tropical support-restricted PINN / tropical_bvp_track.py
Beats tuned baseline
1import numpy as np
2
3META = {
4 "name": "tropical_bvp_local",
5 "domain": "pde",
6 "description": "Manufactured 1D Poisson boundary-value regression: u=sin(pi*x), -u''+u=f, with zero Dirichlet boundaries. Local Taylor support around x=0 is odd powers for the solution.",
7}
8
9def get_dataset(seed, n_train, n_test):
10 rng = np.random.default_rng(int(seed))
11 xtr = rng.uniform(0.0, 1.0, (n_train, 1)).astype(np.float32)
12 xte = np.linspace(0.0, 1.0, n_test, dtype=np.float32).reshape(-1, 1)
13 ytr = np.sin(np.pi * xtr).astype(np.float32)
14 yte = np.sin(np.pi * xte).astype(np.float32)
15 return {"xtr": xtr, "ytr": ytr, "xte": xte, "yte": yte,
16 "task": "regression", "metric": "mse"}