PropensityScore.fit

PropensityScore.fit(X, t, coords, prior=None, noncentred=True)

Build the graph if needed, then sample prior and posterior phases.

Parameters:

• X (ndarray) – Covariate matrix used to predict the treatment.

• t (ndarray) – Observed treatment indicator (0/1).

• coords (dict[str, Any]) – Coordinate names for named dimensions of the model.

• prior (dict[str, list] | None) – Historical compatibility parameter; see build().

• noncentred (bool) – Forwarded to build().

Return type:

DataTree