PyMCModelAdapter.predict

PyMCModelAdapter.predict(X, *, out_of_sample=False, **kwargs)

Predict expected outcomes using the PyMC model.

Parameters:

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

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

• **kwargs (Any) – Additional keyword arguments forwarded to the underlying model.

Returns:

Posterior draws of mu with canonical prediction dimensions.

Return type:

xr.DataArray