smile.base.cart

Class NominalSplit

java.lang.Object

smile.base.cart.Split

smile.base.cart.NominalSplit

public class NominalSplit
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
NominalSplit(LeafNode leaf, int feature, int 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.
NominalNode	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

NominalSplit

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

Constructor.

Method Detail

toNode

public NominalNode 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