TransferFunctionARRegression.fit#
- TransferFunctionARRegression.fit(X, y, coords, treatment_data)[source]#
Fit the Transfer Function AR(1) model.
This method overrides the base PyMCModel.fit() to accept treatment_data.
- Parameters:
X (xr.DataArray) – Baseline design matrix (n_obs, n_baseline_features).
y (xr.DataArray) – Outcome variable (n_obs, 1).
coords (dict) – Coordinate names for PyMC model.
treatment_data (xr.DataArray) – Raw treatment variables (n_obs, n_treatments).
- Returns:
idata – Inference data with posterior, prior predictive, and posterior predictive.
- Return type: