Tropical support-restricted PINN / tropical_bvp_track.py

✓✓ Beats tuned baseline

Raw ⬇ ZIP
 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"}