freealg 0.1.9__tar.gz → 0.1.10__tar.gz

{freealg-0.1.9 → freealg-0.1.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: freealg
-Version: 0.1.9
+Version: 0.1.10
 Summary: Free probability for large matrices
 Keywords: leaderboard bot chat
 Platform: Linux

{freealg-0.1.9 → freealg-0.1.10}/freealg/__init__.py

@@ -7,7 +7,8 @@
 # directory of this source tree.
 
 from .freeform import FreeForm
+from . import distributions
 
-__all__ = ['FreeForm']
+__all__ = ['FreeForm', 'distributions']
 
 from .__version__ import __version__                          # noqa: F401 E402

freealg-0.1.10/freealg/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.1.10"

{freealg-0.1.9 → freealg-0.1.10}/freealg/_plot_util.py

@@ -81,6 +81,59 @@ def plot_fit(psi, x_supp, g_supp, g_supp_approx, support, latex=False,
                                   show_and_save=save_status, verbose=True)
 
 
+# =========
+# auto bins
+# =========
+
+def _auto_bins(array, method='scott', factor=5):
+    """
+    Automatic choice for the number of bins for the histogram of an array.
+
+    Parameters
+    ----------
+
+    array : numpy.array
+        An array for histogram.
+
+    method : {``'freedman'``, ``'scott'``, ``'sturges'``}, default= ``'scott'``
+        Method of choosing number of bins.
+
+    Returns
+    -------
+
+    num_bins : int
+        Number of bins for histogram.
+    """
+
+    if method == 'freedman':
+
+        q75, q25 = numpy.percentile(array, [75, 25])
+        iqr = q75 - q25
+        bin_width = 2 * iqr / (len(array) ** (1/3))
+
+        if bin_width == 0:
+            # Fallback default
+            return
+            num_bins = 100
+        else:
+            num_bins = int(numpy.ceil((array.max() - array.min()) / bin_width))
+
+    elif method == 'scott':
+
+        std = numpy.std(array)
+        bin_width = 3.5 * std / (len(array) ** (1/3))
+        num_bins = int(numpy.ceil((array.max() - array.min()) / bin_width))
+
+    elif method == 'sturges':
+
+        num_bins = int(numpy.ceil(numpy.log2(len(array)) + 1))
+
+    else:
+        raise ValueError('"method" is invalid.')
+
+    return num_bins * factor
+
+
 # ============
 # plot density
 # ============
@@ -96,7 +149,7 @@ def plot_density(x, rho, eig=None, support=None, label='',
 
         if (support is not None) and (eig is not None):
             lam_m, lam_p = support
-            bins = numpy.linspace(lam_m, lam_p, 250)
+            bins = numpy.linspace(lam_m, lam_p, _auto_bins(eig))
             _ = ax.hist(eig, bins, density=True, color='silver',
                         edgecolor='none', label='Histogram')
         else:
@@ -503,7 +556,7 @@ def plot_samples(x, rho, x_min, x_max, samples, latex=False, save=False):
 
         fig, ax = plt.subplots(figsize=(6, 3))
 
-        bins = numpy.linspace(x_min, x_max, samples.size // 15)
+        bins = numpy.linspace(x_min, x_max, _auto_bins(samples))
         _ = ax.hist(samples, bins, density=True, color='silver',
                     edgecolor='none', label='Samples histogram')
         ax.plot(x, rho, color='black', label='Exact density')

{freealg-0.1.9 → freealg-0.1.10}/freealg/_sample.py

@@ -82,7 +82,8 @@ def qmc_sample(x, rho, num_pts, seed=None):
         >>> numpy.allclose(samples.mean(), 0.75, atol=0.02)
     """
 
-    numpy.random.rand(seed)
+    if seed is not None:
+        numpy.random.rand(seed)
 
     quantile = _quantile_func(x, rho)
     engine = qmc.Halton(d=1)

{freealg-0.1.9 → freealg-0.1.10}/freealg/distributions/__init__.py

@@ -6,10 +6,10 @@
 # under the terms of the license found in the LICENSE.txt file in the root
 # directory of this source tree.
 
-from .marchenko_pastur import MarchenkoPastur
-from .wigner import Wigner
-from .kesten_mckay import KestenMcKay
-from .wachter import Wachter
-from .meixner import Meixner
+from ._marchenko_pastur import MarchenkoPastur
+from ._wigner import Wigner
+from ._kesten_mckay import KestenMcKay
+from ._wachter import Wachter
+from ._meixner import Meixner
 
 __all__ = ['MarchenkoPastur', 'Wigner', 'KestenMcKay', 'Wachter', 'Meixner']

freealg-0.1.9/freealg/distributions/kesten_mckay.py → freealg-0.1.10/freealg/distributions/_kesten_mckay.py

@@ -494,7 +494,8 @@ class KestenMcKay(object):
             :class: custom-dark
         """
 
-        numpy.random.seed(seed)
+        if seed is not None:
+            numpy.random.seed(seed)
 
         if x_min is None:
             x_min = self.lam_m

freealg-0.1.9/freealg/distributions/marchenko_pastur.py → freealg-0.1.10/freealg/distributions/_marchenko_pastur.py

@@ -501,7 +501,8 @@ class MarchenkoPastur(object):
             :class: custom-dark
         """
 
-        numpy.random.seed(seed)
+        if seed is not None:
+            numpy.random.seed(seed)
 
         if x_min is None:
             x_min = self.lam_m
@@ -578,7 +579,8 @@ class MarchenkoPastur(object):
             >>> A = mp.matrix(2000)
         """
 
-        numpy.random.seed(seed)
+        if seed is not None:
+            numpy.random.seed(seed)
 
         # Parameters
         m = int(size / self.lam)

freealg-0.1.9/freealg/distributions/meixner.py → freealg-0.1.10/freealg/distributions/_meixner.py

@@ -526,7 +526,8 @@ class Meixner(object):
             :class: custom-dark
         """
 
-        numpy.random.seed(seed)
+        if seed is not None:
+            numpy.random.seed(seed)
 
         if x_min is None:
             x_min = self.lam_m

freealg-0.1.9/freealg/distributions/wachter.py → freealg-0.1.10/freealg/distributions/_wachter.py

@@ -501,7 +501,8 @@ class Wachter(object):
             :class: custom-dark
         """
 
-        numpy.random.seed(seed)
+        if seed is not None:
+            numpy.random.seed(seed)
 
         if x_min is None:
             x_min = self.lam_m
@@ -581,7 +582,8 @@ class Wachter(object):
             >>> A = wa.matrix(2000)
         """
 
-        numpy.random.seed(seed)
+        if seed is not None:
+            numpy.random.seed(seed)
 
         n = size
         m1 = int(self.a * n)

freealg-0.1.9/freealg/distributions/wigner.py → freealg-0.1.10/freealg/distributions/_wigner.py

@@ -478,7 +478,8 @@ class Wigner(object):
             :class: custom-dark
         """
 
-        numpy.random.seed(seed)
+        if seed is not None:
+            numpy.random.seed(seed)
 
         if x_min is None:
             x_min = self.lam_m
@@ -555,7 +556,8 @@ class Wigner(object):
             >>> A = wg.matrix(2000)
         """
 
-        numpy.random.seed(seed)
+        if seed is not None:
+            numpy.random.seed(seed)
 
         # Parameters
         n = size

{freealg-0.1.9 → freealg-0.1.10}/freealg/freeform.py

@@ -499,17 +499,13 @@ class FreeForm(object):
         # Check density is unit mass
         mass = numpy.trapz(rho, x)
         if not numpy.isclose(mass, 1.0, atol=1e-2):
-            # raise RuntimeWarning(f'"rho" is not unit mass. mass: {mass}. ' +
-            #                      r'Set "force=True".')
-            print(f'"rho" is not unit mass. mass: {mass}. Set "force=True".')
+            print(f'"rho" is not unit mass. mass: {mass:>0.3f}. Set ' +
+                  r'"force=True".')
 
         # Check density is positive
         min_rho = numpy.min(rho)
         if min_rho < 0.0 - 1e-3:
-            # raise RuntimeWarning(
-            #         f'"rho" is not positive. min_rho: {min_rho}. Set ' +
-            #         r'"force=True".')
-            print(f'"rho" is not positive. min_rho: {min_rho}. Set ' +
+            print(f'"rho" is not positive. min_rho: {min_rho:>0.3f}. Set ' +
                   r'"force=True".')
 
         if plot:
@@ -631,7 +627,7 @@ class FreeForm(object):
     # stieltjes
     # =========
 
-    def stieltjes(self, x, y, plot=False, latex=False, save=False):
+    def stieltjes(self, x=None, y=None, plot=False, latex=False, save=False):
         """
         Compute Stieltjes transform of the spectral density over a 2D Cartesian
         grid on the complex plane.

{freealg-0.1.9 → freealg-0.1.10}/freealg.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: freealg
-Version: 0.1.9
+Version: 0.1.10
 Summary: Free probability for large matrices
 Keywords: leaderboard bot chat
 Platform: Linux

{freealg-0.1.9 → freealg-0.1.10}/freealg.egg-info/SOURCES.txt

@@ -25,8 +25,8 @@ freealg.egg-info/not-zip-safe
 freealg.egg-info/requires.txt
 freealg.egg-info/top_level.txt
 freealg/distributions/__init__.py
-freealg/distributions/kesten_mckay.py
-freealg/distributions/marchenko_pastur.py
-freealg/distributions/meixner.py
-freealg/distributions/wachter.py
-freealg/distributions/wigner.py
+freealg/distributions/_kesten_mckay.py
+freealg/distributions/_marchenko_pastur.py
+freealg/distributions/_meixner.py
+freealg/distributions/_wachter.py
+freealg/distributions/_wigner.py

freealg-0.1.9/freealg/__version__.py

@@ -1 +0,0 @@
-__version__ = "0.1.9"