Learn R Programming

SDModels (version 1.0.4)

prune.SDTree: 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 SDTree
prune(object, cp, ...)

Value

A pruned SDTree object

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.

Author

Markus Ulmer

See Also

copy

Examples

Run this code
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

Run the code above in your browser using DataLab