CUQIpy 1.3.0.post0.dev104__tar.gz → 1.3.0.post0.dev266__tar.gz

{cuqipy-1.3.0.post0.dev104 → cuqipy-1.3.0.post0.dev266}/CUQIpy.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: CUQIpy
-Version: 1.3.0.post0.dev104
+Version: 1.3.0.post0.dev266
 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.dev104 → cuqipy-1.3.0.post0.dev266}/CUQIpy.egg-info/SOURCES.txt

@@ -45,6 +45,7 @@ cuqi/distribution/_smoothed_laplace.py
 cuqi/distribution/_truncated_normal.py
 cuqi/distribution/_uniform.py
 cuqi/experimental/__init__.py
+cuqi/experimental/_recommender.py
 cuqi/experimental/algebra/__init__.py
 cuqi/experimental/algebra/_ast.py
 cuqi/experimental/algebra/_orderedset.py
@@ -64,7 +65,6 @@ cuqi/experimental/mcmc/_mh.py
 cuqi/experimental/mcmc/_pcn.py
 cuqi/experimental/mcmc/_rto.py
 cuqi/experimental/mcmc/_sampler.py
-cuqi/experimental/mcmc/_utilities.py
 cuqi/geometry/__init__.py
 cuqi/geometry/_geometry.py
 cuqi/implicitprior/__init__.py

{cuqipy-1.3.0.post0.dev104 → cuqipy-1.3.0.post0.dev266}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: CUQIpy
-Version: 1.3.0.post0.dev104
+Version: 1.3.0.post0.dev266
 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.dev104 → cuqipy-1.3.0.post0.dev266}/cuqi/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2025-04-28T12:59:25+0200",
+ "date": "2025-06-23T11:22:39+0300",
  "dirty": false,
  "error": null,
- "full-revisionid": "c62aed4105837bc321e7e325f1b1a2286c659b2f",
- "version": "1.3.0.post0.dev104"
+ "full-revisionid": "0ad94a56cb0776e3b0964f9aebbc01c1b7b41ac1",
+ "version": "1.3.0.post0.dev266"
 }
 '''  # END VERSION_JSON
 

{cuqipy-1.3.0.post0.dev104 → cuqipy-1.3.0.post0.dev266}/cuqi/experimental/__init__.py

@@ -2,3 +2,4 @@
 from . import mcmc
 from . import algebra
 from . import geometry
+from ._recommender import SamplerRecommender

cuqipy-1.3.0.post0.dev266/cuqi/experimental/_recommender.py

@@ -0,0 +1,200 @@
+import cuqi
+import inspect
+import numpy as np
+
+# This import makes suggest_sampler easier to read
+import cuqi.experimental.mcmc as samplers 
+
+
+class SamplerRecommender(object):
+    """
+    This class can be used to automatically choose a sampler.
+
+    Parameters
+    ----------
+    target: Density or JointDistribution
+        Distribution to get sampler recommendations for.
+
+    exceptions: list[cuqi.experimental.mcmc.Sampler], *optional*
+        Samplers not to be recommended.
+    """
+
+    def __init__(self, target:cuqi.density.Density, exceptions = []):
+        self._target = target
+        self._exceptions = exceptions
+        self._create_ordering()
+
+    @property
+    def target(self) -> cuqi.density.Density:
+        """ Return the target Distribution. """
+        return self._target
+    
+    @target.setter
+    def target(self, value:cuqi.density.Density):
+        """ Set the target Distribution. Runs validation of the target. """
+        if value is None:
+            raise ValueError("Target needs to be of type cuqi.density.Density.")
+        self._target = value
+
+    def _create_ordering(self):
+        """
+        Every element in the ordering consists of a tuple:
+        (
+            Sampler: Class
+            boolean: additional conditions on the target
+            parameters: additional parameters to be passed to the sampler once initialized
+        )
+        """
+        number_of_components = np.sum(self._target.dim)
+
+        self._ordering = [
+            # Direct and Conjugate samplers
+            (samplers.Direct, True, {}),
+            (samplers.Conjugate, True, {}),
+            (samplers.ConjugateApprox, True, {}),
+            # Specialized samplers
+            (samplers.LinearRTO, True, {}),
+            (samplers.RegularizedLinearRTO, True, {}),
+            (samplers.UGLA, True, {}),
+            # Gradient.based samplers (Hamiltonian and Langevin)
+            (samplers.NUTS, True, {}),
+            (samplers.MALA, True, {}),
+            (samplers.ULA, True, {}),
+            # Gibbs and Componentwise samplers
+            (samplers.HybridGibbs, True, {"sampling_strategy" : self.recommend_HybridGibbs_sampling_strategy(as_string = False)}),
+            (samplers.CWMH, number_of_components <= 100, {"scale" : 0.05*np.ones(number_of_components),
+                            "initial_point" : 0.5*np.ones(number_of_components)}),
+            # Proposal based samplers
+            (samplers.PCN, True, {"scale" : 0.02}),
+            (samplers.MH, number_of_components <= 1000, {}),
+        ]
+
+    @property
+    def ordering(self):
+        """ Returns the ordered list of recommendation rules used by the recommender. """
+        return self._ordering
+    
+    def valid_samplers(self, as_string = True):
+        """
+        Finds all possible samplers that can be used for sampling from the target distribution.
+
+        Parameters
+        ----------
+
+        as_string : boolean
+            Whether to return the name of the sampler as a string instead of instantiating a sampler. *Optional*
+
+        """
+
+        all_samplers = [(name, cls) for name, cls in inspect.getmembers(cuqi.experimental.mcmc, inspect.isclass) if issubclass(cls, cuqi.experimental.mcmc.Sampler)]
+        valid_samplers = []
+
+        for name, sampler in all_samplers:
+            try:
+                sampler(self.target)
+                valid_samplers += [name if as_string else sampler]
+            except:
+                pass
+
+        # Need a separate case for HybridGibbs
+        if self.valid_HybridGibbs_sampling_strategy() is not None:
+            valid_samplers += [cuqi.experimental.mcmc.HybridGibbs.__name__ if as_string else cuqi.experimental.mcmc.HybridGibbs]
+
+        return valid_samplers
+    
+
+    def valid_HybridGibbs_sampling_strategy(self, as_string = True):
+        """
+            Find all possible sampling strategies to be used with the HybridGibbs sampler.
+            Returns None if no sampler could be suggested for at least one conditional distribution.
+
+            Parameters
+            ----------
+
+            as_string : boolean
+                Whether to return the name of the samplers in the sampling strategy as a string instead of instantiating samplers. *Optional*
+        
+
+        """
+
+        if not isinstance(self.target, cuqi.distribution.JointDistribution):
+            return None
+
+        par_names = self.target.get_parameter_names()
+
+        valid_samplers = dict()
+        for par_name in par_names:
+            conditional_params = {par_name_: np.ones(self.target.dim[i]) for i, par_name_ in enumerate(par_names) if par_name_ != par_name}
+            conditional = self.target(**conditional_params)
+
+            recommender = SamplerRecommender(conditional)
+            samplers = recommender.valid_samplers(as_string)
+            if len(samplers) == 0:
+                return None
+            
+            valid_samplers[par_name] = samplers
+
+        return valid_samplers
+    
+
+    def recommend(self, as_string = False):
+        """
+            Suggests a possible sampler that can be used for sampling from the target distribution.
+            Return None if no sampler could be suggested.
+
+            Parameters
+            ----------
+
+            as_string : boolean
+                Whether to return the name of the sampler as a string instead of instantiating a sampler. *Optional*
+
+        """
+
+        valid_samplers = self.valid_samplers(as_string = False)
+
+        for suggestion, flag, values in self._ordering:
+            if flag and (suggestion in valid_samplers) and (suggestion not in self._exceptions):
+                # Sampler found
+                if as_string:
+                    return suggestion.__name__
+                else:
+                    return suggestion(self.target, **values)
+
+        # No sampler can be suggested
+        raise ValueError("Cannot suggest any sampler. Either the provided distribution is incorrectly defined or there are too many exceptions provided.")
+
+    def recommend_HybridGibbs_sampling_strategy(self, as_string = False):
+        """
+            Suggests a possible sampling strategy to be used with the HybridGibbs sampler.
+            Returns None if no sampler could be suggested for at least one conditional distribution.
+
+            Parameters
+            ----------
+
+            target : `cuqi.distribution.JointDistribution`
+                The target distribution get a sampling strategy for.
+            
+            as_string : boolean
+                Whether to return the name of the samplers in the sampling strategy as a string instead of instantiating samplers. *Optional*
+        
+        """
+
+        if not isinstance(self.target, cuqi.distribution.JointDistribution):
+            return None
+
+        par_names = self.target.get_parameter_names()
+
+        suggested_samplers = dict()
+        for par_name in par_names:
+            conditional_params = {par_name_: np.ones(self.target.dim[i]) for i, par_name_ in enumerate(par_names) if par_name_ != par_name}
+            conditional = self.target(**conditional_params)
+
+            recommender = SamplerRecommender(conditional, exceptions = self._exceptions.copy())
+            sampler = recommender.recommend(as_string = as_string)
+
+            if sampler is None:
+                return None
+            
+            suggested_samplers[par_name] = sampler
+
+        return suggested_samplers

{cuqipy-1.3.0.post0.dev104 → cuqipy-1.3.0.post0.dev266}/cuqi/experimental/geometry/_productgeometry.py

@@ -172,10 +172,10 @@ class _ProductGeometry(Geometry):
         return tuple(funvecs)
         
     
-    def __repr__(self) -> str:
+    def __repr__(self, pad="") -> str:
         """Representation of the product geometry."""
         string = "{}(".format(self.__class__.__name__) + "\n"
         for g in self.geometries:
-            string += "\t{}\n".format(g.__repr__())
-        string += ")"
+            string += pad + "    {}\n".format(g.__repr__())
+        string += pad + ")"
         return string

{cuqipy-1.3.0.post0.dev104 → cuqipy-1.3.0.post0.dev266}/cuqi/experimental/mcmc/__init__.py

@@ -120,4 +120,3 @@ from ._gibbs import HybridGibbs
 from ._conjugate import Conjugate
 from ._conjugate_approx import ConjugateApprox
 from ._direct import Direct
-from ._utilities import find_valid_samplers