fit_on_train: Fit estimator_plan on training sample

In microsoft/CausalGrid: Analysis of Subgroups

Fit estimator_plan on training sample

Description

Initializes the estimator (if necessary) on the train sample.

Usage

fit_on_train(obj, X_tr, y_tr, d_tr, cv_folds, verbosity = 0, dim_cat = c())

Arguments

obj	an EstimatorPlan object
X_tr	N_trxK matrix
y_tr	N_tr-vector
d_tr	NULL, N_tr-vector, or N_trxM matrix (if multiple treatments)
cv_folds	CV folds
verbosity	verbosity
dim_cat	vector of dimensions that are categorical

Value

Updated EstimatorPlan object

microsoft/CausalGrid documentation built on Aug. 25, 2022, 9:30 a.m.