CUQIpy 1.3.0.post0.dev395__tar.gz → 1.3.0.post0.dev401__tar.gz

{cuqipy-1.3.0.post0.dev395 → cuqipy-1.3.0.post0.dev401}/CUQIpy.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: CUQIpy
-Version: 1.3.0.post0.dev395
+Version: 1.3.0.post0.dev401
 Summary: Computational Uncertainty Quantification for Inverse problems in Python
 Maintainer-email: "Nicolai A. B. Riis" <nabr@dtu.dk>, "Jakob S. Jørgensen" <jakj@dtu.dk>, "Amal M. Alghamdi" <amaal@dtu.dk>, Chao Zhang <chaz@dtu.dk>
 License:                                  Apache License

{cuqipy-1.3.0.post0.dev395 → cuqipy-1.3.0.post0.dev401}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: CUQIpy
-Version: 1.3.0.post0.dev395
+Version: 1.3.0.post0.dev401
 Summary: Computational Uncertainty Quantification for Inverse problems in Python
 Maintainer-email: "Nicolai A. B. Riis" <nabr@dtu.dk>, "Jakob S. Jørgensen" <jakj@dtu.dk>, "Amal M. Alghamdi" <amaal@dtu.dk>, Chao Zhang <chaz@dtu.dk>
 License:                                  Apache License

{cuqipy-1.3.0.post0.dev395 → cuqipy-1.3.0.post0.dev401}/cuqi/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2025-09-19T16:37:46+0300",
+ "date": "2025-09-21T15:00:42+0200",
  "dirty": false,
  "error": null,
- "full-revisionid": "2cf72ec9af9af17dad4bb3870ee20d303376de24",
- "version": "1.3.0.post0.dev395"
+ "full-revisionid": "f08eb2ae4f6f5209d4a06a4e031bc61bb746ce01",
+ "version": "1.3.0.post0.dev401"
 }
 '''  # END VERSION_JSON
 

{cuqipy-1.3.0.post0.dev395 → cuqipy-1.3.0.post0.dev401}/cuqi/experimental/mcmc/_sampler.py

@@ -203,13 +203,16 @@ class Sampler(ABC):
 
         self.set_state(state)
 
-    def sample(self, Ns, batch_size=0, sample_path='./CUQI_samples/') -> 'Sampler':
+    def sample(self, Ns, Nt=1, batch_size=0, sample_path='./CUQI_samples/') -> 'Sampler':
         """ Sample Ns samples from the target density.
 
         Parameters
         ----------
         Ns : int
             The number of samples to draw.
+        
+        Nt : int, optional, default=1
+            The thinning interval. If Nt >= 1, every Nt'th sample is stored. The larger Nt, the fewer samples are stored.
 
         batch_size : int, optional
             The batch size for saving samples to disk. If 0, no batching is used. If positive, samples are saved to disk in batches of the specified size.
@@ -233,7 +236,8 @@ class Sampler(ABC):
 
             # Store samples
             self._acc.append(acc)
-            self._samples.append(self.current_point)
+            if (Nt > 0) and ((idx + 1) % Nt == 0):
+                self._samples.append(self.current_point)
 
             # display acc rate at progress bar
             pbar.set_postfix_str(f"acc rate: {np.mean(self._acc[-1-idx:]):.2%}")
@@ -248,7 +252,7 @@ class Sampler(ABC):
         return self
     
 
-    def warmup(self, Nb, tune_freq=0.1) -> 'Sampler':
+    def warmup(self, Nb, Nt=1, tune_freq=0.1) -> 'Sampler':
         """ Warmup the sampler by drawing Nb samples.
 
         Parameters
@@ -256,6 +260,9 @@ class Sampler(ABC):
         Nb : int
             The number of samples to draw during warmup.
 
+        Nt : int, optional, default=1
+            The thinning interval. If Nt >= 1, every Nt'th sample is stored. The larger Nt, the fewer samples are stored.
+            
         tune_freq : float, optional
             The frequency of tuning. Tuning is performed every tune_freq*Nb samples.
 
@@ -278,7 +285,8 @@ class Sampler(ABC):
 
             # Store samples
             self._acc.append(acc)
-            self._samples.append(self.current_point)
+            if (Nt > 0) and ((idx + 1) % Nt == 0):
+                self._samples.append(self.current_point)
 
             # display acc rate at progress bar
             pbar.set_postfix_str(f"acc rate: {np.mean(self._acc[-1-idx:]):.2%}")