split_forest: Split Node in Random Forest EAT

View source: R/splitRFEAT.R

split_forestR Documentation

Split Node in Random Forest EAT

Description

This function gets the variable and split value to be used in estimEAT, selects the best split, node indexes and leaf list.

Usage

split_forest(data, tree, leaves, t, x, y, numStop, arrayK)

Arguments

data

Data to be used.

tree

List structure with the tree nodes.

leaves

List with leaf nodes or pending expansion nodes.

t

Node which is being split.

x

Column input indexes in data.

y

Column output indexes in data.

numStop

Minimum number of observations on a node to be split.

arrayK

Column input indexes in data selected by s_mtry.

Value

Leaves and tree lists updated with the new child nodes.


eat documentation built on Jan. 10, 2023, 5:13 p.m.