PyMCModel.build_model#

PyMCModel.build_model(X, y, coords)[source]#

Construct the PyMC model graph.

Subclasses must override this method to declare priors, deterministic nodes, and the likelihood for the model.

Parameters:
  • X (DataArray) – Input features with dimensions ["obs_ind", "coeffs"].

  • y (DataArray) – Target variable with dimensions ["obs_ind", "treated_units"].

  • coords (dict[str, Any] | None) – Mapping of named dimensions to coordinate labels for the underlying pm.Model.

Raises:

NotImplementedError – Always, when called on the base class.

Return type:

None