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 and Description
OrdinalSplit(LeafNode leaf, int feature, double value, double score, int lo, int hi, int trueCount, int falseCount, java.util.function.IntPredicate predicate) Constructor.

Method Summary

Modifier and Type	Method and Description
java.util.function.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 Detail

OrdinalSplit

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

Constructor.

Method Detail

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 java.util.function.IntPredicate predicate()

Description copied from class: Split

Returns the lambda that tests on the split feature.

Specified by:

predicate in class Split