folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #2 from fAndreuzzi/patch-1

Browse Source 
Use `np.polynomial.chebyshev.chebroots` instead of `np.roots`
This commit is contained in:
Nicola Demo
2022-02-17 09:44:08 +01:00
committed by GitHub
parent 1b26f2b725 66dfa79428
commit 765b7108dc
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pina/chebyshev.py
View File

@@ -1,6 +1,7 @@
from mpmath import chebyt, chop, taylor
import numpy as np import numpy as np
def chebyshev_roots(n): def chebyshev_roots(n):
""" Return the roots of *n* Chebyshev polynomials (between [-1, 1]) """ """ Return the roots of *n* Chebyshev polynomials (between [-1, 1]) """
return np.roots(chop(taylor(lambda x: chebyt(n, x), 0, n))[::-1]) coefficents = np.zeros(n+1)
coefficents[-1] = 1
return np.polynomial.chebyshev.chebroots(coefficents)

3
setup.py
View File

@@ -14,8 +14,7 @@ VERSION = meta['__version__']
KEYWORDS = 'physics-informed neural-network' KEYWORDS = 'physics-informed neural-network'
REQUIRED = [ REQUIRED = [
'future', 'numpy', 'matplotlib', 'torch', 'mpmath' 'future', 'numpy', 'matplotlib', 'torch'
] ]
EXTRAS = { EXTRAS = {