Description Usage Arguments Value Examples
View source: R/recursive_tree.R
Decision tree in a recursive way
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
data |
the dataset including the response |
Y |
the name of the response |
modele |
(optional) vector of names of covariates allowed in the tree |
kill |
vector of the names to kill (variables won't be used in the tree) |
index |
to give a number to the plot |
verbose |
boolean to print the tree parameters |
plot |
boolean to plot the tree |
main |
the main title if plot=TRUE |
sub |
the subtitle (if NULL it is automatically added) |
lang |
the language for the automatic subtitle in the plot |
all |
Logical. If TRUE, all nodes are labeled, otherwise just terminal nodes. |
digits |
number of digits for legend of the leaves |
returns the tree as an "rpart" object and the modele as a vector of the names of the covariates the tree could have used (to give as an input of the function).
modele |
vector of the names of the covariates the tree could have used |
tree |
the regression tree as an "rpart" object |
1 2 3 4 5 6 | data <- mtcars
main = "Regression tree of cars consumption (in mpg)"
mytree = recursive_tree(data = data, Y = "mpg", main = main, verbose = FALSE)
# want to try without cylinder and disp
mytree2 = recursive_tree(data = data, Y = "mpg", kill = c("cyl", "disp"),
modele = mytree$modele, main = main, verbose = FALSE)
|
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.