smile.classification

Enum DecisionTree.SplitRule

java.lang.Object

java.lang.Enum<DecisionTree.SplitRule>

smile.classification.DecisionTree.SplitRule

All Implemented Interfaces:

Serializable, Comparable<DecisionTree.SplitRule>

Enclosing class:

DecisionTree

public static enum DecisionTree.SplitRule
extends Enum<DecisionTree.SplitRule>

The criterion to choose variable to split instances.

Enum Constant Summary

Enum Constants

Enum Constant and Description
ENTROPY Used by the ID3, C4.5 and C5.0 tree generation algorithms.
GINI Used by the CART algorithm, Gini impurity is a measure of how often a randomly chosen element from the set would be incorrectly labeled if it were randomly labeled according to the distribution of labels in the subset.

Method Summary

Methods

Modifier and Type	Method and Description
static DecisionTree.SplitRule	valueOf(String name) Returns the enum constant of this type with the specified name.
static DecisionTree.SplitRule[]	values() Returns an array containing the constants of this enum type, in the order they are declared.

Methods inherited from class java.lang.Enum

clone, compareTo, equals, finalize, getDeclaringClass, hashCode, name, ordinal, toString, valueOf

Methods inherited from class java.lang.Object

getClass, notify, notifyAll, wait, wait, wait

Enum Constant Detail

GINI

public static final DecisionTree.SplitRule GINI

Used by the CART algorithm, Gini impurity is a measure of how often a randomly chosen element from the set would be incorrectly labeled if it were randomly labeled according to the distribution of labels in the subset. Gini impurity can be computed by summing the probability of each item being chosen times the probability of a mistake in categorizing that item. It reaches its minimum (zero) when all cases in the node fall into a single target category.

ENTROPY

public static final DecisionTree.SplitRule ENTROPY

Used by the ID3, C4.5 and C5.0 tree generation algorithms.

Method Detail

values

public static DecisionTree.SplitRule[] values()

Returns an array containing the constants of this enum type, in the order they are declared. This method may be used to iterate over the constants as follows:

for (DecisionTree.SplitRule c : DecisionTree.SplitRule.values())
    System.out.println(c);

Returns:

an array containing the constants of this enum type, in the order they are declared

valueOf

public static DecisionTree.SplitRule valueOf(String name)

Returns the enum constant of this type with the specified name. The string must match exactly an identifier used to declare an enum constant in this type. (Extraneous whitespace characters are not permitted.)

Parameters:

name - the name of the enum constant to be returned.

Returns:

the enum constant with the specified name

Throws:

IllegalArgumentException - if this enum type has no constant with the specified name

NullPointerException - if the argument is null