PyAutoFit: Classy Probabilistic Programming
Installation Guide | readthedocs | Introduction on Colab | HowToFit
PyAutoFit is a Python based probabilistic programming language for model fitting and Bayesian inference of large datasets.
The basic PyAutoFit API allows us a user to quickly compose a probabilistic model and fit it to data via a log likelihood function, using a range of non-linear search algorithms (e.g. MCMC, nested sampling).
Users can then set up PyAutoFit scientific workflow, which enables streamlined modeling of small datasets with tools to scale up to large datasets.
PyAutoFit supports advanced statistical methods, most notably a big data framework for Bayesian hierarchical analysis.
Getting Started
The following links are useful for new starters:
- The PyAutoFit readthedocs, which includes an installation guide and an overview of PyAutoFit's core features.
- The introduction Jupyter Notebook on Colab, where you can try PyAutoFit in a web browser (without installation).
- The autofit_workspace GitHub repository, which includes example scripts demonstrating PyAutoFit's features.
- The standalone HowToFit repository, a series of Jupyter notebook lectures which give new users a step-by-step introduction to PyAutoFit.
Support
Support for installation issues, help with Fit modeling and using PyAutoFit is available by raising an issue on the GitHub issues page.
We also offer support on the PyAutoFit Slack channel, where we also provide the latest updates on PyAutoFit. Slack is invitation-only, so if you'd like to join send an email requesting an invite.
HowToFit
For users less familiar with Bayesian inference and scientific analysis you may wish to read through the HowToFits lectures. These teach you the basic principles of Bayesian inference, with the content pitched at undergraduate level and above.
The lectures are available in the standalone HowToFit repository.
API Overview
To illustrate the PyAutoFit API, we use an illustrative toy model of fitting a one-dimensional Gaussian to
noisy 1D data. Here's the data (black) and the model (red) we'll fit:
We define our model, a 1D Gaussian by writing a Python class using the format below:
class Gaussian:
def __init__(
self,
centre=0.0, # <- PyAutoFit recognises these
normalization=0.1, # <- constructor arguments are
sigma=0.01, # <- the Gaussian's parameters.
):
self.centre = centre
self.normalization = normalization
self.sigma = sigma
"""
An instance of the Gaussian class will be available during model fitting.
This method will be used to fit the model to data and compute a likelihood.
"""
def model_data_from(self, xvalues):
transformed_xvalues = xvalues - self.centre
return (self.normalization / (self.sigma * (2.0 * np.pi) ** 0.5)) * \
np.exp(-0.5 * (transformed_xvalues / self.sigma) ** 2.0)
PyAutoFit recognises that this Gaussian may be treated as a model component whose parameters can be fitted for via a non-linear search like emcee.
To fit this Gaussian to the data we create an Analysis object, which gives PyAutoFit the data and a
log_likelihood_function describing how to fit the data with the model:
class Analysis(af.Analysis):
def __init__(self, data, noise_map):
self.data = data
self.noise_map = noise_map
def log_likelihood_function(self, instance):
"""
The 'instance' that comes into this method is an instance of the Gaussian class
above, with the parameters set to values chosen by the non-linear search.
"""
print("Gaussian Instance:")
print("Centre = ", instance.centre)
print("normalization = ", instance.normalization)
print("Sigma = ", instance.sigma)
"""
We fit the ``data`` with the Gaussian instance, using its
"model_data_from" function to create the model data.
"""
xvalues = np.arange(self.data.shape[0])
model_data = instance.model_data_from(xvalues=xvalues)
residual_map = self.data - model_data
chi_squared_map = (residual_map / self.noise_map) ** 2.0
log_likelihood = -0.5 * sum(chi_squared_map)
return log_likelihood
We can now fit our model to the data using a non-linear search:
model = af.Model(Gaussian)
analysis = Analysis(data=data, noise_map=noise_map)
emcee = af.Emcee(nwalkers=50, nsteps=2000)
result = emcee.fit(model=model, analysis=analysis)
The result contains information on the model-fit, for example the parameter samples, maximum log likelihood
model and marginalized probability density functions.
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file autofit-2026.9.8.1.tar.gz.
File metadata
- Download URL: autofit-2026.9.8.1.tar.gz
- Upload date:
- Size: 7.6 MB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.0.1 CPython/3.12.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
639cc92dd2ce14e7d5eb87a05b8e5a1eadd744cd347f9d3c8a0c254a2543e46d
|
|
| MD5 |
24288f74bfcb4710f782266f149035d1
|
|
| BLAKE2b-256 |
b6133c8d9c5a9c2a4e6151695e37bc2bcb923451136a087537f76e80d30f4075
|
File details
Details for the file autofit-2026.9.8.1-py3-none-any.whl.
File metadata
- Download URL: autofit-2026.9.8.1-py3-none-any.whl
- Upload date:
- Size: 7.7 MB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.0.1 CPython/3.12.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
219a2d6ea66b8969df45ac0c806aa3ed97bed99c522c6e0d961e1a5e1a16eece
|
|
| MD5 |
d413d47962016a8be647336aae61fc6d
|
|
| BLAKE2b-256 |
7ee27a12434ca5266fd471e6a13f8b8df677697937efee7c4f1456955bfbfb3c
|
