PyMCModelAdapter.predict

PyMCModelAdapter.predict(X, *, coords=None, out_of_sample=False)

Predict expected outcomes using the PyMC model.

Parameters:

• X (Any) – Predictor matrix for which to generate predictions.

• coords (dict[str, Any] | None) – Coordinate metadata for the PyMC model.

• out_of_sample (bool) – Whether predictions are out-of-sample.

Returns:

Posterior draws of mu with canonical prediction dimensions.

Return type:

xr.DataArray