PIEModel.build_model#

PIEModel.build_model(X, y, **kwargs)[source]#

Build the PyMC model graph.

Parameters:
Xpd.DataFrame

Campaign feature matrix. Only the pre_determined_features and post_determined_features columns are used — any other columns (e.g. an id column) are ignored. Columns with non-numeric dtype are label-encoded automatically.

ypd.Series or np.ndarray

Measured incrementality (one value per campaign).