smile.base.cart

Class Split

java.lang.Object

smile.base.cart.Split

Direct Known Subclasses:

NominalSplit, OrdinalSplit

public abstract class Split
extends java.lang.Object

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

Field Summary

Fields

Modifier and Type	Field and Description
static java.util.Comparator<Split>	comparator

Constructor Summary

Constructors

Constructor and Description
Split(LeafNode leaf, int feature, double score, int lo, int hi, int trueCount, int falseCount) Constructor.

Method Summary

Modifier and Type	Method and Description
abstract java.util.function.IntPredicate	predicate() Returns the lambda that tests on the split feature.
abstract InternalNode	toNode(Node trueChild, Node falseChild) Returns an internal node with the feature, value, and score of this split.
java.lang.String	toString()

Methods inherited from class java.lang.Object

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

Field Detail

comparator

public static java.util.Comparator<Split> comparator

Constructor Detail

Split

public Split(LeafNode leaf,
             int feature,
             double score,
             int lo,
             int hi,
             int trueCount,
             int falseCount)

Constructor.

Method Detail

toNode

public abstract InternalNode toNode(Node trueChild,
                                    Node falseChild)

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

Parameters:

trueChild - the child node of true branch.

falseChild - the child node of false branch.

Returns:

an internal node

predicate

public abstract java.util.function.IntPredicate predicate()

Returns the lambda that tests on the split feature.

toString

public java.lang.String toString()

Overrides:

toString in class java.lang.Object