changed bins in estimate_tau

Author: rsexton2

basicrta/gibbs.py

@@ -274,7 +274,7 @@ def cluster(self, method="GaussianMixture", **kwargs):
         setattr(self.processed_results, 'indicator', pindicator)
         setattr(self.processed_results, 'labels', all_labels)
 
-    def process_gibbs(self):
+    def process_gibbs(self, show=True):
         r"""
         Process the samples collected from the Gibbs sampler.
         :meth:`process_gibbs` can be called multiple times to check the
@@ -298,7 +298,7 @@ def process_gibbs(self):
         lmode = stats.mode(lens).mode
 
         self.cluster(n_init=117, n_components=lmode)
-        labels, presorts = mixture_and_plot(self)
+        labels, presorts = mixture_and_plot(self, show=show)
         setattr(self.processed_results, 'labels', labels)
         setattr(self.processed_results, 'indicator',
                 self.processed_results.indicator[:, presorts])
@@ -712,8 +712,8 @@ def estimate_tau(self):
 
         taus = 1 / rp.rates[rp.labels == index]
         ci = confidence_interval(taus)
-        bins = 15
-        h = np.histogram(taus, bins=bins)
+        bins = 10
+        h = np.histogram(taus, bins='sqrt')
         indmax = h[0].argmax()
         val = 0.5 * (h[1][:-1][indmax] + h[1][1:][indmax])
         return [ci[0], val, ci[1]]

basicrta/util.py

@@ -692,7 +692,7 @@ def extract_data(gibbs):
     return data, train_inds
 
 def mixture_and_plot(gibbs, scale=2, sparse=1, remove_noise=False, wlim=None,
-                     rlim=None, **kwargs):
+                     rlim=None, show=True, **kwargs):
     from scipy import stats
     from matplotlib.ticker import MaxNLocator
 
@@ -1192,10 +1192,11 @@ def mixture_and_plot(gibbs, scale=2, sparse=1, remove_noise=False, wlim=None,
                          bbox_inches='tight')
             figp.savefig(f"{basename}_validate.{suffix}",
                          bbox_inches='tight')
-
-    figa.show()
-    figt.show()
-    figp.show()
+    
+    if show:
+        figa.show()
+        figt.show()
+        figp.show()
     return all_labels, presorts