display acc rate of the samples from the current sample/warmup

cuqi/experimental/mcmc/_sampler.py

@@ -221,7 +221,7 @@ def sample(self, Ns, batch_size=0, sample_path='./CUQI_samples/') -> 'Sampler':
 
         # Draw samples
         pbar = tqdm(range(Ns), "Sample: ")
-        for _ in pbar:
+        for idx in pbar:
 
             # Perform one step of the sampler
             acc = self.step()
@@ -231,7 +231,7 @@ def sample(self, Ns, batch_size=0, sample_path='./CUQI_samples/') -> 'Sampler':
             self._samples.append(self.current_point)
 
             # display acc rate at progress bar
-            pbar.set_postfix_str(f"acc rate: {np.mean(self._acc):.2%}")
+            pbar.set_postfix_str(f"acc rate: {np.mean(self._acc[-1-idx:]):.2%}")
 
             # Add sample to batch
             if batch_size > 0:
@@ -279,7 +279,7 @@ def warmup(self, Nb, tune_freq=0.1) -> 'Sampler':
             self._samples.append(self.current_point)
 
             # display acc rate at progress bar
-            pbar.set_postfix_str(f"acc rate: {np.mean(self._acc):.2%}")
+            pbar.set_postfix_str(f"acc rate: {np.mean(self._acc[-1-idx:]):.2%}")
 
             # Call callback function if specified
             self._call_callback(self.current_point, len(self._samples)-1)