sparse-ir 0.97.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

sparse_ir/__init__.py

@@ -10,9 +10,9 @@ intermediate representation of correlation functions.  It provides:
  - basis functions and singular values are accurate to full precision
  - routines for sparse sampling
 """
-__copyright__ = "2020-2022 Markus Wallerberger, Hiroshi Shinaoka, and others"
+__copyright__ = "2020-2024 Markus Wallerberger, Hiroshi Shinaoka, and others"
 __license__ = "MIT"
-__version__ = "0.97.0"
+__version__ = "1.1.0"
 
 from .kernel import RegularizedBoseKernel, LogisticKernel
 from .sve import compute as compute_sve, SVEResult

sparse_ir/sampling.py

@@ -20,13 +20,70 @@ class AbstractSampling:
             |________________|      fit        |___________________|
 
     """
-    def evaluate(self, al, axis=None):
-        """Evaluate the basis coefficients at the sparse sampling points"""
+    def evaluate(self, al, axis=None, *, points=None):
+        """Evaluate the basis coefficients at sampling points.
+
+        Arguments:
+            al (array):
+                Array where the `l`-th item along `axis` corresponds to the
+                `l`-th basis coefficient
+            axis (integer):
+                Axis or dimension of `al` along which to evaluate the function.
+                Defaults to the last, i.e., rightmost axis.
+            points (vector):
+                Points on which the results should be evaluated.  Defaults
+                to the sampling points for which the sampling objects was
+                created.
+
+        Return:
+            Array where the `n`-th item along `axis` corresponds to the
+            value on the `n`-th sampling point (or value on `point[n]`, if
+            given.)
+
+        Note:
+            If `points` is given, a new sampling is created at each invocation,
+            which can result in a performance hit.  Consider caching sampling
+            objects or simply using the `.u()` and `.uhat()` methods of the
+            underlying basis.
+        """
+        if points is not None:
+            return self._for_sampling_points(points).evaluate(al, axis)
+
         return self.matrix.matmul(al, axis)
 
-    def fit(self, ax, axis=None):
-        """Fit basis coefficients from the sparse sampling points"""
-        return self.matrix.lstsq(ax, axis)
+    def fit(self, ax, axis=None, *, points=None):
+        """Fit the basis coefficients from the sampling points.
+
+        Arguments:
+            ax (array):
+                Array where the `n`-th item along `axis` corresponds to the
+                value on the `n`-th sampling point (or value on `point[n]`, if
+                given.)
+            axis (integer):
+                Axis or dimension of `ax` along which to fit the function.
+                Defaults to the last, i.e., rightmost axis.
+            points (vector):
+                Points on which the `ax` is given.  Defaults to the sampling
+                points for which the sampling objects was created.
+
+        Return:
+            Array where the `l`-th item along `axis` corresponds to the
+            `l`-th basis coefficient
+
+        Note:
+            If `points` is given, a new sampling is created at each invocation,
+            which can result in a performance hit.  Consider caching sampling
+            objects.
+        """
+        if points is not None:
+            return self._for_sampling_points(points).fit(ax, axis)
+
+        matrix = self.matrix
+        if self.basis.is_well_conditioned and not (matrix.cond <= 1e8):
+            warn(f"Sampling matrix is poorly conditioned "
+                 f"(kappa = {matrix.cond:.2g})", ConditioningWarning)
+
+        return matrix.lstsq(ax, axis)
 
     @property
     def cond(self):
@@ -48,6 +105,9 @@ class AbstractSampling:
         """Basis instance"""
         raise NotImplementedError()
 
+    def _for_sampling_points(self, x):
+        raise RuntimeError("Changing sampling points is not possible")
+
 
 class TauSampling(AbstractSampling):
     """Sparse sampling in imaginary time.
@@ -68,10 +128,6 @@ class TauSampling(AbstractSampling):
         self._sampling_points = sampling_points
         self._matrix = DecomposedMatrix(matrix)
 
-        if basis.is_well_conditioned and self._matrix.cond > 1e8:
-            warn(f"Sampling matrix is poorly conditioned "
-                 f"(kappa = {self._matrix.cond:.2g})", ConditioningWarning)
-
     @property
     def basis(self): return self._basis
 
@@ -86,6 +142,10 @@ class TauSampling(AbstractSampling):
         """Sampling points in (reduced) imaginary time"""
         return self._sampling_points
 
+    def _for_sampling_points(self, x):
+        x = np.asarray(x)
+        return TauSampling(self._basis, x)
+
 
 class MatsubaraSampling(AbstractSampling):
     """Sparse sampling in Matsubara frequencies.
@@ -125,10 +185,6 @@ class MatsubaraSampling(AbstractSampling):
         else:
             self._matrix = DecomposedMatrix(matrix)
 
-        if basis.is_well_conditioned and self._matrix.cond > 1e8:
-            warn(f"Sampling matrix is poorly conditioned "
-                 f"(kappa = {self._matrix.cond:.2g})", ConditioningWarning)
-
     @property
     def basis(self): return self._basis
 
@@ -148,6 +204,11 @@ class MatsubaraSampling(AbstractSampling):
         """Sampling points as (reduced) Matsubara frequencies"""
         return self._sampling_points
 
+    def _for_sampling_points(self, x):
+        x = np.asarray(x)
+        return MatsubaraSampling(self._basis, x,
+                                 positive_only=self._positive_only)
+
 
 class DecomposedMatrix:
     """Matrix in SVD decomposed form for fast and accurate fitting.
@@ -314,6 +375,13 @@ class SplitDecomposedMatrix:
 
 
 class ConditioningWarning(RuntimeWarning):
+    """Warns about a poorly conditioned problem.
+
+    This warning is issued if the library detects a poorly conditioned fitting
+    problem.  This essentially means there is a high degree of ambiguity in how
+    to choose the solution.  One must therefore expect to lose significant
+    precision in the parameter values.
+    """
     pass
 
 

{sparse_ir-0.97.0.dist-info → sparse_ir-1.1.0.dist-info}/METADATA

@@ -1,14 +1,14 @@
 Metadata-Version: 2.1
 Name: sparse-ir
-Version: 0.97.0
+Version: 1.1.0
 Summary: intermediate representation (IR) basis for electronic propagator
 Home-page: https://github.com/SpM-lab/sparse-ir
 Author: ['Markus Wallerberger', 'Hiroshi Shinaoka', 'Kazuyoshi Yoshimi', 'Junya Otsuki', 'Chikano Naoya']
 Author-email: markus.wallerberger@tuwien.ac.at
 License: UNKNOWN
-Keywords: irbasis
+Keywords: irbasis many-body propagator svd
 Platform: UNKNOWN
-Classifier: Development Status :: 4 - Beta
+Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
 Classifier: Topic :: Scientific/Engineering :: Physics
@@ -16,6 +16,7 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3
 Description-Content-Type: text/x-rst
+License-File: LICENSE.txt
 Requires-Dist: numpy
 Requires-Dist: scipy
 Requires-Dist: setuptools
@@ -143,7 +144,7 @@ useful in your research, please consider citing the following papers:
 
  - Hiroshi Shinaoka et al., `Phys. Rev. B 96, 035147`_  (2017)
  - Jia Li et al., `Phys. Rev. B 101, 035144`_ (2020)
- - Markus Wallerberger et al., `arXiv 2206.11762`_ (2022)
+ - Markus Wallerberger et al., `SoftwareX 21, 101266`_ (2023)
 
 If you are discussing sparse sampling in your research specifically, please
 also consider citing an independently discovered, closely related approach, the
@@ -152,7 +153,7 @@ MINIMAX isometry method (Merzuk Kaltak and Georg Kresse,
 
 .. _Phys. Rev. B 96, 035147: https://doi.org/10.1103/PhysRevB.96.035147
 .. _Phys. Rev. B 101, 035144: https://doi.org/10.1103/PhysRevB.101.035144
-.. _arXiv 2206.11762: https://doi.org/10.48550/arXiv.2206.11762
+.. _SoftwareX 21, 101266: https://doi.org/10.1016/j.softx.2022.101266
 .. _Phys. Rev. B 101, 205145: https://doi.org/10.1103/PhysRevB.101.205145
 
 

{sparse_ir-0.97.0.dist-info → sparse_ir-1.1.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-sparse_ir/__init__.py,sha256=nfmgA_ldvgUIS0U_ziVxgE2exlR7lnkZoEjm89Bg8A4,817
+sparse_ir/__init__.py,sha256=BzVg_maIuofnQArELnsb3Gdi4LnHa7OPt7prTjdbnnM,816
 sparse_ir/_gauss.py,sha256=9Ou38SfucUjY83o8Tz62s-gXfWDC8QEHBjiUzcbWaY4,9621
 sparse_ir/_roots.py,sha256=ARpyYhhI5gR1AT1zDgezVgCxnHEaJwGF38jEv6ebarA,4351
 sparse_ir/_util.py,sha256=93s7tGcXxz3mwHsBQmAh-6EQ26xLbUeFHVtlnzQxRps,3004
@@ -10,11 +10,11 @@ sparse_ir/basis_set.py,sha256=zLD51j9X3iuXfX2nzmn5yRRn7S_GOOyCRFqb05OMDXI,2869
 sparse_ir/dlr.py,sha256=X6orDjvsOAsFrZsCh8vfedu0_SmTvIpSQmGabfI9pi8,5361
 sparse_ir/kernel.py,sha256=2HO77HTOK2WJKizDgr1nmmyhd1mGYT2WvKrFwLiMhjM,18487
 sparse_ir/poly.py,sha256=XbNjgFaofE9zSvPPPYmsDVLLea2k8CSEopfxx-e48ko,18359
-sparse_ir/sampling.py,sha256=nrGKNJ11edfTvreq6ta99aZTTYCqxwGtEShQJZagc8g,11130
+sparse_ir/sampling.py,sha256=AR6ONMwKjQXVvmulwCNk93J-1OmNS8fYutXCLdfReCA,13801
 sparse_ir/svd.py,sha256=iQwpBCFSPJeKqhAIlufzz9Ybpm1dq382Km3TU-UvNuc,3375
 sparse_ir/sve.py,sha256=pVUKg_cLphx-lRCfrDJXrpkow97vjIgvvSrxA7lhzzg,15179
-sparse_ir-0.97.0.dist-info/LICENSE.txt,sha256=3tGlA0QNYsfjETaQqJO0Ixne5PQ16PNVJiDcGsgHER0,1079
-sparse_ir-0.97.0.dist-info/METADATA,sha256=O-YzP67aCgZHZmPCEw11fscNbN_gvPvYZaTBkZfpS2s,6385
-sparse_ir-0.97.0.dist-info/WHEEL,sha256=g4nMs7d-Xl9-xC9XovUrsDHGXt-FT0E17Yqo92DEfvY,92
-sparse_ir-0.97.0.dist-info/top_level.txt,sha256=UsscWAzJg7fKo9qmIwW8jnG7CAfhFzWYBOTXVySzuA0,10
-sparse_ir-0.97.0.dist-info/RECORD,,
+sparse_ir-1.1.0.dist-info/LICENSE.txt,sha256=3tGlA0QNYsfjETaQqJO0Ixne5PQ16PNVJiDcGsgHER0,1079
+sparse_ir-1.1.0.dist-info/METADATA,sha256=MIFpjO8gtK_A2NzU6g4SNYhZ-OKdUaNZqTpSD3V_VIU,6458
+sparse_ir-1.1.0.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+sparse_ir-1.1.0.dist-info/top_level.txt,sha256=UsscWAzJg7fKo9qmIwW8jnG7CAfhFzWYBOTXVySzuA0,10
+sparse_ir-1.1.0.dist-info/RECORD,,

{sparse_ir-0.97.0.dist-info → sparse_ir-1.1.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.34.2)
+Generator: bdist_wheel (0.37.1)
 Root-Is-Purelib: true
 Tag: py3-none-any