prune.SDTree {SDModels} | R Documentation |
Prune an SDTree
Description
Removes all nodes that did not improve the loss by more than cp times the initial loss. Either by themselves or by one of their successors. Note that the tree is pruned in place. If you intend to keep the original tree, make a copy of it before pruning.
Usage
## S3 method for class 'SDTree'
prune(object, cp, ...)
Arguments
object |
an SDTree object |
cp |
Complexity parameter, the higher the value the more nodes are pruned. |
... |
Further arguments passed to or from other methods. |
Value
A pruned SDTree object
Author(s)
Markus Ulmer
See Also
Examples
set.seed(1)
X <- matrix(rnorm(10 * 20), nrow = 10)
Y <- rnorm(10)
tree <- SDTree(x = X, y = Y)
pruned_tree <- prune(copy(tree), 0.2)
tree
pruned_tree
[Package SDModels version 1.0.13 Index]