BaseDelayedSaturatedMMM.predict_posterior

BaseDelayedSaturatedMMM.predict_posterior(X_pred, extend_idata=True, combined=True, **kwargs)

Generate posterior predictive samples on unseen data.

Parameters:

• X_pred (array-like if sklearn is available, otherwise array, shape (n_pred, n_features)) – The input data used for prediction.

• extend_idata (Boolean determining whether the predictions should be added to inference data object.) – Defaults to True.

• combined (Combine chain and draw dims into sample. Won't work if a dim named sample already exists.) – Defaults to True.

• **kwargs (Additional arguments to pass to sample_posterior_predictive method)

Returns:

y_pred – Posterior predictive samples for each input X_pred

Return type:

DataArray, shape (n_pred, chains * draws) if combined is True, otherwise (chains, draws, n_pred)