mcf_functions.ModifiedCausalForest.train

ModifiedCausalForest.train(data_df)

Build the modified causal forest on the training data.

Parameters

data_df (DataFrame) – Data used to compute the causal forest. It must contain information about outcomes, treatment, and features.

Returns

• tree_df (DataFrame) – Dataset used to build the forest.

• fill_y_df (DataFrame) – Dataset used to populate the forest with outcomes.

• outpath (String) – Location of directory in which output is saved.