Skip to main content

Bayesian A/B testing for proportions

Project description

BayesProp Logo

Bayesian A/B Testing for Proportions

PyPI Downloads Python License: MIT Tests codecov

A Python package for Bayesian hypothesis testing of success-rate differences in any Bernoulli-like experiment, using analytic and approximate inference methods — lightweight and dependency-lean (no PyMC, Pyro, Stan, or other heavy probabilistic-programming frameworks required). Input data can be binary (0/1) or real-valued on (0, 1) — continuous scores are automatically binarized at a configurable threshold. Typical applications include comparing treatments, groups, items, model variants, or any two conditions whose outcomes can be expressed as proportions. Please check out our Getting Started guide for installation and quick examples.

Features

  • Effect-size inference for proportions — estimate and test the difference in success rates for both paired and non-paired samples
  • Hierarchical logistic regression — optionally place Inverse-Gamma hyperpriors on the prior variances so the model learns the prior scales from data, reducing sensitivity to prior choice (Jeffreys–Lindley robustness)
  • Savage–Dickey Bayes Factor — test a point-null hypothesis ('treatment effect / difference is zero') without fitting a separate null model
  • Posterior of the null & ROPE — quantify the posterior mass inside a Region of Practical Equivalence for nuanced decisions beyond simple reject/accept
  • Posterior predictive checks — assess model fit by comparing observed data to data simulated from the posterior
  • Bayes Factor Design Analysis (BFDA) — plan sample sizes to reach a target level of evidence before running the experiment
  • Sequential / streaming design — update the posterior batch-by-batch as data arrive and stop early once the Bayes factor crosses an upper or lower threshold (SequentialNonPairedBayesPropTest, SequentialPairedBayesPropTest)
  • Operating-characteristic analysiscalibrated-Bayes frequentist evaluation of the chosen decision rule: three-way decision rates (reject / accept / inconclusive), Type-I sweep over the baseline rate, 95 % credible-interval coverage, and the sequential stopping-time distribution, with matched-α Fisher's exact (non-paired) or McNemar exact (paired) baselines overlaid. Pre-built Monte-Carlo harness in bayesprop.utils.operation_characteristics and …_paired, plus turnkey notebooks for both designs
  • Publication-ready plots — posterior distributions, predictive checks, Savage–Dickey density-ratio plots, BFDA power curves, sequential BF₁₀ trajectories, and OC diagnostic plots (with Wilson Monte-Carlo bands) out of the box

Models

All paired methods are accessible through a single unified facadePairedBayesPropTest(method=…) — that dispatches to the chosen inference backend.

Model Class / method Method When to use
Non-paired Beta–Bernoulli NonPairedBayesPropTest Conjugate Beta posteriors per arm; P(B>A) by quadrature, Δ summaries by Monte Carlo Independent groups, exact & fast
Paired Logistic (Laplace) PairedBayesPropTest(method="laplace") MAP + Laplace (fixed or hierarchical IG hyperpriors) Paired scores, fast, default
Paired Logistic (Pólya–Gamma) PairedBayesPropTest(method="pg") Exact Gibbs sampling (fixed or hierarchical IG hyperpriors) Paired scores, small n, exact posterior
Paired Bayesian Bootstrap PairedBayesPropTest(method="bootstrap") Nonparametric — Dirichlet weights on paired differences Paired scores, no prior elicitation, ROPE-driven (no Savage–Dickey BF)

Quick start

import numpy as np
from bayesprop.resources.bayes_paired import PairedBayesPropTest

# Paired binary data (y_A[i] and y_B[i] refer to the same item)
y_A = np.array([1,1,0,1,1,0,1,1,1,1,1,1,1,0,1,1,1,0,1,1])     # 16/20 = 0.80
y_B = np.array([0,1,0,0,1,0,0,1,0,0,1,0,1,0,0,1,0,0,0,0])     #  6/20 = 0.30

# Fit posterior & summarise
model = PairedBayesPropTest(seed=42).fit(y_A, y_B)

s = model.summary
print(f"θ_A = {s.theta_A_mean:.4f},  θ_B = {s.theta_B_mean:.4f}")
print(f"Mean Δ (θ_A − θ_B) = {s.mean_delta:+.4f}")
print(f"95% CI = [{s.ci_95.lower:.4f}, {s.ci_95.upper:.4f}]")
print(f"P(A > B) = {s.p_A_greater_B:.4f}")

# ── Unified decision ─────────────────────────────────────────────────
d = model.decide()
bf = d.bayes_factor

print("\n--- Unified Decision ---")
print(f"  Bayes Factor: BF_10 = {bf.BF_10:.2f}{bf.decision}")
print(f"  Posterior Null: P(H0|D) = {d.posterior_null.p_H0:.4f}{d.posterior_null.decision}")
print(f"  ROPE: {d.rope.decision} ({d.rope.pct_in_rope:.1%} in ROPE)")

# Plots
model.plot_posteriors()
model.plot_posterior_delta()
model.plot_savage_dickey()

Installation

pip install bayesprop

Or with uv:

uv add bayesprop

For development (from source):

git clone https://github.com/AVoss84/bayesProp.git
cd bayesprop
uv venv --python 3.13
uv sync
source .venv/bin/activate

Dependencies

  • Python ≥ 3.13
  • numpy, scipy, matplotlib, pandas
  • pydantic (v2)
  • polyagamma

References

  • Gelman, A., Carlin, J. B., Stern, H. S., Dunson, D. B., Vehtari, A. & Rubin, D. B. (2013). Bayesian Data Analysis (3rd ed.). Chapman & Hall/CRC.
  • Kruschke, J. K. (2018). Rejecting or accepting parameter values in Bayesian estimation. Advances in Methods and Practices in Psychological Science, 1(2), 270–280.
  • Polson, N. G., Scott, J. G. & Windle, J. (2013). Bayesian inference for logistic models using Pólya–Gamma latent variables. JASA, 108(504), 1339–1349.
  • Rubin, D. B. (1981). The Bayesian Bootstrap. The Annals of Statistics, 9(1), 130–134.
  • Schönbrodt, F. D. & Wagenmakers, E.-J. (2018). Bayes factor design analysis: Planning for compelling evidence. Psychonomic Bulletin & Review, 25(1), 128–142.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

bayesprop-0.1.1.2.tar.gz (106.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

bayesprop-0.1.1.2-py3-none-any.whl (89.8 kB view details)

Uploaded Python 3

File details

Details for the file bayesprop-0.1.1.2.tar.gz.

File metadata

  • Download URL: bayesprop-0.1.1.2.tar.gz
  • Upload date:
  • Size: 106.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for bayesprop-0.1.1.2.tar.gz
Algorithm Hash digest
SHA256 b360f6189e67f53a61c1b7cd3c82bd66d1080f4b18c0829edca28cc903085d11
MD5 060a6ba12ee407958d36e792466b030b
BLAKE2b-256 db8e5fb8d63571644e91ad26aaa4fdb01430f43acf3da9ee104bd58b2beba3e0

See more details on using hashes here.

Provenance

The following attestation bundles were made for bayesprop-0.1.1.2.tar.gz:

Publisher: publish_pypi.yml on AVoss84/bayesProp

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file bayesprop-0.1.1.2-py3-none-any.whl.

File metadata

  • Download URL: bayesprop-0.1.1.2-py3-none-any.whl
  • Upload date:
  • Size: 89.8 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for bayesprop-0.1.1.2-py3-none-any.whl
Algorithm Hash digest
SHA256 9a2cad77e77006125ad21f3af39070794d0b21568a1e8493577a5e17a6239433
MD5 6efb6b8f0baca9e190f8bcd1e51e1f73
BLAKE2b-256 b1594b266e95f3da9e3e5dcb3752ce1bb1dd8e24c4fde3bc2652877d86b190fd

See more details on using hashes here.

Provenance

The following attestation bundles were made for bayesprop-0.1.1.2-py3-none-any.whl:

Publisher: publish_pypi.yml on AVoss84/bayesProp

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page