dualbounds.dist_reg.DistReg.feature_transform¶

DistReg.feature_transform(W: array, X: array, Z: array | None = None)[source]¶

Transforms the features before feeding them to the base model.

Parameters:¶

X : np.array¶

(n, p)-shaped array of covariates.

W : np.array¶

n-length array of treatment/exposure indicators.

Z : np.array¶

Optional n-length array of binary instruments for the instrumental variables setting.

Returns:¶

features – (n, d)-shaped array of features.

Return type:¶

np.array