Fit estimator_plan on training sample

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

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