model_builder#

Base classes containing primitives and high-level API for model building, fitting, saving, and loading.

Functions

create_idata_accessor(value, message)

Create a property accessor for an InferenceData object.

create_sample_kwargs(sampler_config, ...)

Create the dictionary of keyword arguments for pm.sample.

requires_model(func)

Ensure that the model is built before calling a method.

Classes

ModelBuilder([model_config, sampler_config])

Base class for building PyMC-Marketing models.

ModelIO()

Mixin to handle saving and loading of models.

RegressionModelBuilder([model_config, ...])

ModelBuilder class providing an easy-to-use API similar to scikit-learn for regression models.

Exceptions

DifferentModelError

Error raised when a model loaded is different than one saved.