API

Modules

bass	Bass diffusion model for product adoption forecasting.
causal_utils	A moducle with utilities for causal reasoning and discovery.
clv	CLV models and utilities.
customer_choice	Customer Choice module.
data	Data adaptors for PyMC-Marketing.
hsgp_kwargs	Class to store and validate keyword argument for the Hilbert Space Gaussian Process (HSGP) components.
metrics	Evaluation Metrics.
mlflow	MLflow logging utilities for PyMC models.
mmm	Marketing Mix Models (MMM).
model_builder	Base classes containing primitives and high-level API for model building, fitting, saving, and loading.
model_config	Model configuration utilities.
model_graph	Functions to manipulate PyMC models as graphs.
paths	Paths for the project.
plot	Plot distributions stored in xarray.DataArray across coordinates.
prior	Class that represents a prior distribution.
pytensor_utils	PyTensor utility functions.
serialization	Unified serialization infrastructure for pymc-marketing.
serialization_migration	Migration tool for old-format serialized models.
special_priors	Specialized priors that behave like the Prior class.
utils	Utility functions for PyMC-Marketing.