prune.SDTree: Prune an SDTree

In SDModels: Spectrally Deconfounded Models

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

copy

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

SDModels documentation built on April 11, 2025, 5:50 p.m.