smile.base.cart

Class DecisionNode

java.lang.Object

smile.base.cart.LeafNode

smile.base.cart.DecisionNode

All Implemented Interfaces:

java.io.Serializable, Node

public class DecisionNode
extends LeafNode

A leaf node in decision tree.

See Also:

Serialized Form

Field Summary

Fields inherited from class smile.base.cart.LeafNode

size

Constructor Summary

Constructors

Constructor and Description
DecisionNode(int[] count) Constructor.

Method Summary

Modifier and Type	Method and Description
int[]	count() Returns the number of node samples in each class.
double	deviance() Returns the deviance of node.
static double	deviance(int[] count, double[] prob) Returns the deviance of node.
java.lang.String	dot(smile.data.type.StructType schema, smile.data.type.StructField response, int id) Returns a dot representation for visualization.
boolean	equals(java.lang.Object o)
double	impurity(SplitRule rule) Returns the impurity of node.
static double	impurity(SplitRule rule, int size, int[] count) Returns the impurity of samples.
int	output() Returns the predicted value.
double[]	posteriori(double[] prob) Returns the class probability.
static double[]	posteriori(int[] count, double[] prob) Returns the class probability.
int[]	toString(smile.data.type.StructType schema, smile.data.type.StructField response, InternalNode parent, int depth, java.math.BigInteger id, java.util.List<java.lang.String> lines) Adds the string representation (R's rpart format) to a collection.

Methods inherited from class smile.base.cart.LeafNode

depth, leafs, merge, predict, size

Methods inherited from class java.lang.Object

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

Constructor Detail

DecisionNode

public DecisionNode(int[] count)

Constructor.

Parameters:

count - the number of node samples in each class.

Method Detail

output

public int output()

Returns the predicted value.

count

public int[] count()

Returns the number of node samples in each class.

deviance

public double deviance()

Description copied from interface: Node

Returns the deviance of node.

dot

public java.lang.String dot(smile.data.type.StructType schema,
                            smile.data.type.StructField response,
                            int id)

Description copied from interface: Node

Returns a dot representation for visualization.

Parameters:

schema - the schema of data

response - the schema of response variable

id - node id

toString

public int[] toString(smile.data.type.StructType schema,
                      smile.data.type.StructField response,
                      InternalNode parent,
                      int depth,
                      java.math.BigInteger id,
                      java.util.List<java.lang.String> lines)

Description copied from interface: Node

Adds the string representation (R's rpart format) to a collection.

Parameters:

schema - the schema of data

response - the schema of response variable

parent - the parent node

depth - the depth of node in the tree. The root node is at depth 0.

id - node id

lines - the collection of node's string representation.

Returns:

the sample count of each class for decision tree; single element array [node size] for regression tree.

impurity

public double impurity(SplitRule rule)

Returns the impurity of node.

Returns:

the impurity of node

impurity

public static double impurity(SplitRule rule,
                              int size,
                              int[] count)

Returns the impurity of samples.

Parameters:

size - the number of samples.

count - the number of samples in each class.

Returns:

the impurity of node

equals

public boolean equals(java.lang.Object o)

Overrides:

equals in class java.lang.Object

posteriori

public double[] posteriori(double[] prob)

Returns the class probability.

posteriori

public static double[] posteriori(int[] count,
                                  double[] prob)

Returns the class probability.

deviance

public static double deviance(int[] count,
                              double[] prob)

Returns the deviance of node.