Package smile.base.cart

Class OrdinalSplit

java.lang.Object

smile.base.cart.Split

smile.base.cart.OrdinalSplit

public class OrdinalSplit
extends Split

The data about of a potential split for a leaf node.

Field Summary

Fields inherited from class smile.base.cart.Split

comparator

Constructor Summary

Constructors

Constructor	Description
OrdinalSplit(LeafNode leaf, int feature, double value, double score, int lo, int hi, int trueCount, int falseCount, IntPredicate predicate)	Constructor.

Method Summary

Modifier and Type	Method	Description
IntPredicate	predicate()	Returns the lambda that tests on the split feature.
OrdinalNode	toNode(Node trueChild, Node falseChild)	Returns an internal node with the feature, value, and score of this split.

Methods inherited from class smile.base.cart.Split

toString

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Details

OrdinalSplit

public OrdinalSplit(LeafNode leaf,
 int feature,
 double value,
 double score,
 int lo,
 int hi,
 int trueCount,
 int falseCount,
 IntPredicate predicate)

Constructor.

Parameters:

leaf - the node to split.

feature - the index of feature column.

value - the split value.

score - the split score.

lo - the lower bound of sample index in the node.

hi - the upper bound of sample index in the node.

trueCount - the number of samples in true branch child.

falseCount - the number of samples false branch child.

predicate - the lambda returns true if the sample passes the test on the split feature.

Method Details

toNode

public OrdinalNode toNode(Node trueChild,
 Node falseChild)

Description copied from class: Split

Returns an internal node with the feature, value, and score of this split.

Specified by:

toNode in class Split

Parameters:

trueChild - the child node of true branch.

falseChild - the child node of false branch.

Returns:

an internal node

predicate

public IntPredicate predicate()

Description copied from class: Split

Returns the lambda that tests on the split feature.

Specified by:

predicate in class Split

Returns:

the lambda that tests on the split feature.