PIEModel.sample_posterior_predictive

PIEModel.sample_posterior_predictive(X, extend_idata=True, combined=True, **kwargs)

Sample posterior predictive draws and return in the original target scale.

Parameters:

Xpd.DataFrame

Campaign feature matrix for prediction. Same schema as training X (no target column required).

extend_idatabool

Whether to attach predictions to self.idata. Defaults to True.

combinedbool

Whether to combine chain and draw dims into a single sample dim. Defaults to True.

Returns:

xarray.Dataset

Posterior predictive draws in the original target scale, containing self.output_var. The output variable has shape (obs, sample) when combined=True.