From eb10123f544d6c673f83b9b4969afdc7e9c64c96 Mon Sep 17 00:00:00 2001 From: "J.-C. Jiang" Date: Thu, 12 Dec 2024 17:34:24 +0800 Subject: [PATCH] Fix feynman.py equation 114 --- kan/feynman.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/kan/feynman.py b/kan/feynman.py index 6cc55e96..3b25e555 100644 --- a/kan/feynman.py +++ b/kan/feynman.py @@ -697,7 +697,7 @@ def get_feynman_dataset(name): if name == 'Jackson 2.11' or name == 114: symbol = q, eps, y, Ve, d = symbols('q epsilon y V_e d') expr = q/(4*pi*eps*y**2)*(4*pi*eps*Ve*d-q*d*y**3/(y**2-d**2)**2) - f = lambda x: x[:,[0]]/(4*tpi*x[:,[1]]*x[:,x[:,[2]]]**2)*(4*tpi*x[:,[1]]*x[:,[3]]*x[:,[4]]-x[:,[0]]*x[:,[4]]*x[:,[2]]**3/(x[:,[2]]**2-x[:,[4]]**2)**2) + f = lambda x: x[:,[0]]/(4*tpi*x[:,[1]]*x[:,[2]]]**2)*(4*tpi*x[:,[1]]*x[:,[3]]*x[:,[4]]-x[:,[0]]*x[:,[4]]*x[:,[2]]**3/(x[:,[2]]**2-x[:,[4]]**2)**2) ranges = [[0,1],[0.5,2],[1,2],[0,1],[0,1]] if name == 'Jackson 3.45' or name == 115: @@ -736,4 +736,4 @@ def get_feynman_dataset(name): f = lambda x: tpi*x[:,[0]]**2*x[:,[1]]**2/x[:,[2]]**2/x[:,[3]]**2*(x[:,[4]]/x[:,[5]])**2*(x[:,[4]]/x[:,[5]]+x[:,[5]]/x[:,[4]]-torch.sin(x[:,[6]])**2) ranges = [[0,1],[0,1],[0.5,2],[0.5,2],[0.5,2],[0.5,2],[0,2*tpi]] - return symbol, expr, f, ranges \ No newline at end of file + return symbol, expr, f, ranges