aspire-inference 0.1.0a8__py3-none-any.whl → 0.1.0a9__py3-none-any.whl

aspire/__init__.py

@@ -6,6 +6,7 @@ import logging
 from importlib.metadata import PackageNotFoundError, version
 
 from .aspire import Aspire
+from .samples import Samples
 
 try:
     __version__ = version("aspire")
@@ -16,4 +17,5 @@ logging.getLogger(__name__).addHandler(logging.NullHandler())
 
 __all__ = [
     "Aspire",
+    "Samples",
 ]

aspire/samplers/smc/blackjax.py

@@ -220,7 +220,9 @@ class BlackJAXSMC(SMCSampler):
 
         elif algorithm == "nuts":
             # Initialize step size and mass matrix if not provided
-            inverse_mass_matrix = self.sampler_kwargs["inverse_mass_matrix"]
+            inverse_mass_matrix = self.sampler_kwargs.get(
+                "inverse_mass_matrix"
+            )
             if inverse_mass_matrix is None:
                 inverse_mass_matrix = jax.numpy.eye(self.dims)
 
@@ -258,8 +260,13 @@ class BlackJAXSMC(SMCSampler):
             z_final = final_states.position
 
             # Calculate acceptance rates
-            acceptance_rates = jax.numpy.mean(all_infos.is_accepted, axis=1)
-            mean_acceptance = jax.numpy.mean(acceptance_rates)
+            try:
+                acceptance_rates = jax.numpy.mean(
+                    all_infos.is_accepted, axis=1
+                )
+                mean_acceptance = jax.numpy.mean(acceptance_rates)
+            except AttributeError:
+                mean_acceptance = np.nan
 
         elif algorithm == "hmc":
             # Initialize HMC sampler
@@ -294,8 +301,13 @@ class BlackJAXSMC(SMCSampler):
 
             final_states, all_infos = jax.vmap(init_and_sample)(keys, z_jax)
             z_final = final_states.position
-            acceptance_rates = jax.numpy.mean(all_infos.is_accepted, axis=1)
-            mean_acceptance = jax.numpy.mean(acceptance_rates)
+            try:
+                acceptance_rates = jax.numpy.mean(
+                    all_infos.is_accepted, axis=1
+                )
+                mean_acceptance = jax.numpy.mean(acceptance_rates)
+            except AttributeError:
+                mean_acceptance = np.nan
 
         else:
             raise ValueError(f"Unsupported algorithm: {algorithm}")

aspire/transforms.py

@@ -332,6 +332,7 @@ class CompositeTransform(BaseTransform):
             prior_bounds=self.prior_bounds,
             bounded_to_unbounded=self.bounded_to_unbounded,
             bounded_transform=self.bounded_transform,
+            affine_transform=self.affine_transform,
             device=self.device,
             xp=xp or self.xp,
             eps=self.eps,

{aspire_inference-0.1.0a8.dist-info → aspire_inference-0.1.0a9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aspire-inference
-Version: 0.1.0a8
+Version: 0.1.0a9
 Summary: Accelerate Sequential Posterior Inference via REuse
 Author-email: "Michael J. Williams" <michaeljw1@googlemail.com>
 License: MIT
@@ -39,6 +39,12 @@ Dynamic: license-file
 
 # aspire: Accelerated Sequential Posterior Inference via REuse
 
+[![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.15658747.svg)](https://doi.org/10.5281/zenodo.15658747)
+[![PyPI](https://img.shields.io/pypi/v/aspire-inference)](https://pypi.org/project/aspire-inference/)
+[![Documentation Status](https://readthedocs.org/projects/aspire/badge/?version=latest)](https://aspire.readthedocs.io/en/latest/?badge=latest)
+![tests](https://github.com/mj-will/aspire/actions/workflows/tests.yml/badge.svg)
+
+
 aspire is a framework for reusing existing posterior samples to obtain new results at a reduced cost.
 
 ## Installation
@@ -52,6 +58,45 @@ pip install aspire-inference
 **Important:** the name of `aspire` on PyPI is `aspire-inference` but once installed
 the package can be imported and used as `aspire`.
 
+## Quickstart
+
+```python
+import numpy as np
+from aspire import Aspire, Samples
+
+# Define a log-likelihood and log-prior
+def log_likelihood(samples):
+    x = samples.x
+    return -0.5 * np.sum(x**2, axis=-1)
+
+def log_prior(samples):
+    return -0.5 * np.sum(samples.x**2, axis=-1)
+
+# Create the initial samples
+init = Samples(np.random.normal(size=(2_000, 4)))
+
+# Define the aspire object
+aspire = Aspire(
+    log_likelihood=log_likelihood,
+    log_prior=log_prior,
+    dims=4,
+    parameters=[f"x{i}" for i in range(4)],
+)
+
+# Fit the normalizing flow
+aspire.fit(init, n_epochs=20)
+
+# Sample the posterior
+posterior = aspire.sample_posterior(
+    sampler="smc",
+    n_samples=500,
+    sampler_kwargs=dict(n_steps=100),
+)
+
+# Plot the posterior distribution
+posterior.plot_corner()
+```
+
 ## Documentation
 
 See the [documentation on ReadTheDocs][docs].

{aspire_inference-0.1.0a8.dist-info → aspire_inference-0.1.0a9.dist-info}/RECORD

@@ -1,9 +1,9 @@
-aspire/__init__.py,sha256=45R0xWaLg0aJEPK5zoTK0aIek0KOwpHwQWS1jLCDhIE,365
+aspire/__init__.py,sha256=B2IETvlpB0oBh57prRYLdi8jB5yFGw8qVviGdf1NcnE,409
 aspire/aspire.py,sha256=M5o-QxLthE_5daa1trgUfWxPz-g4rmpEUKimKosw4lw,17400
 aspire/history.py,sha256=l_j-riZKbTWK7Wz9zvvD_mTk9psNCKItiveYhr_pYv8,4313
 aspire/plot.py,sha256=oXwUDOb_953_ADm2KLk41JIfpE3JeiiQiSYKvUVwLqw,1423
 aspire/samples.py,sha256=nVQULOr19lQVyGhitI8EgDdCDGx9sExPomQBJrV4rTc,19237
-aspire/transforms.py,sha256=W2JL045zkczvdBVXXeSn2c8aKyX8ASxFKgWU2cJHufk,24922
+aspire/transforms.py,sha256=CHrfPQHEyHQ9I0WWiAgUplWwxYypZsD7uCYIHUbSFtY,24974
 aspire/utils.py,sha256=pj8O0chqfP6VS8bpW0wCw8W0P5JNQKvWRz1Rg9AYIhg,22525
 aspire/flows/__init__.py,sha256=GUZToPVNJoTwULpbeW10UijfQukNrILoAQ_ubeq7G3w,2110
 aspire/flows/base.py,sha256=5UWKAiXDXLJ6Sg6a380ajLrGFaZSQyOnFEihQiiA4ko,2237
@@ -18,11 +18,11 @@ aspire/samplers/importance.py,sha256=opn_jY-V8snUz0JztLBtnaTT3WfrZ5OSpHBV5WAuM3M
 aspire/samplers/mcmc.py,sha256=ihHgzqvSoy1oxdFBjyqNUbCuRX7CqWjlshCUZcgEL5E,5151
 aspire/samplers/smc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aspire/samplers/smc/base.py,sha256=66f_ORUvcKRqMIW35qjhUc-c0PFuY87lJa91MpSaTZI,10729
-aspire/samplers/smc/blackjax.py,sha256=4L4kgRKlaWl-knTWXXzdJTh-zZBh5BTpy5GaLDzT8Sc,11803
+aspire/samplers/smc/blackjax.py,sha256=IcTguAETiPmgFofmVW2GN40P5HBIxkmyd2VR8AU8f4k,12115
 aspire/samplers/smc/emcee.py,sha256=Wm0vvAlCcRhJMBt7_fU2ZnjDb8SN8jgUOTXLzNstRpA,2516
 aspire/samplers/smc/minipcn.py,sha256=ju1gcgyKHjodLEACPdL3eXA9ai8ZJ9_LwitD_Gmf1Rc,2765
-aspire_inference-0.1.0a8.dist-info/licenses/LICENSE,sha256=DN-eRtBfS9dZyT0Ds0Mdn2Y4nb-ZQ7h71vpASYBm5k4,1076
-aspire_inference-0.1.0a8.dist-info/METADATA,sha256=Or73qfnx3KsPNSeaNTJ70t56GME2r9qACM03aSW40t8,1965
-aspire_inference-0.1.0a8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aspire_inference-0.1.0a8.dist-info/top_level.txt,sha256=9FRIYEl2xwVFG7jSOBHsElHQ0y3_4fq01Cf4_OyMQn8,7
-aspire_inference-0.1.0a8.dist-info/RECORD,,
+aspire_inference-0.1.0a9.dist-info/licenses/LICENSE,sha256=DN-eRtBfS9dZyT0Ds0Mdn2Y4nb-ZQ7h71vpASYBm5k4,1076
+aspire_inference-0.1.0a9.dist-info/METADATA,sha256=DCcTGComTNt9UFlbUwGPf58itBX6Y-1-edVys0pb2RQ,3187
+aspire_inference-0.1.0a9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aspire_inference-0.1.0a9.dist-info/top_level.txt,sha256=9FRIYEl2xwVFG7jSOBHsElHQ0y3_4fq01Cf4_OyMQn8,7
+aspire_inference-0.1.0a9.dist-info/RECORD,,