Fix cayley_loss function.

pulquero · Jul 25, 2024 · 799419e · 799419e
1 parent 0979ea6
commit 799419e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
  name="sgw_tools",
- version="3.1",
+ version="3.1.1",
  author="Mark Hale",
  license="MIT",
  description="Spectral graph wavelet tools",

diff --git a/sgw_tools/approximations.py b/sgw_tools/approximations.py
@@ -134,5 +134,4 @@ def cayley_loss(h, f, m):
  z = util.cayley_transform(h*G.e)
  y = f.evaluate(G.e).squeeze()
  p = np.polynomial.polynomial.Polynomial.fit(z, y, m, domain=[0, G.lmax], window=[0, G.lmax])
- p.coef[0] = np.real(p.coef[0])
- return np.linalg.norm(y - p(z))
+ return np.linalg.norm(y - p(z).real)