Description Usage Arguments Examples
View source: R/cf_helper_functions.R
This function takes in a dataframe with features and an outcome. It produces predictions for a test set, and plots them against the features.
1 2 | causal_forest_plots(df, y, x_var, exclude = NULL, split_r = 0.9,
seed_1 = 24010, seed_2 = 39028, seed_3 = 45829)
|
df |
is your dataframe |
y |
is the outcome being predicted. |
x_var |
is the specific feature to plot against that outcome |
exclude |
takes any variables you want to omit from the sample. |
split_r |
is the training/test proportion split you want. Defaults to .9. |
seed_1 |
and those that follow allow you to see the random partition for reproducibility. |
1 |
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.