smile.base.cart

Interface Node

All Superinterfaces:

java.io.Serializable

All Known Implementing Classes:

DecisionNode, InternalNode, LeafNode, NominalNode, OrdinalNode, RegressionNode

public interface Node
extends java.io.Serializable

CART tree node.

Method Summary

Modifier and Type	Method and Description
int	depth() Returns the maximum depth of the tree -- the number of nodes along the longest path from this node down to the farthest leaf node.
double	deviance() 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.
int	leafs() Returns the number of leaf nodes in the subtree.
Node	merge() Try to merge the children nodes and return a leaf node.
LeafNode	predict(smile.data.Tuple x) Evaluate the tree over an instance.
int	size() Returns the number of samples in the node.
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.

Method Detail

predict

LeafNode predict(smile.data.Tuple x)

Evaluate the tree over an instance.

dot

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

Returns a dot representation for visualization.

Parameters:

schema - the schema of data

response - the schema of response variable

id - node id

size

int size()

Returns the number of samples in the node.

leafs

int leafs()

Returns the number of leaf nodes in the subtree.

depth

int depth()

Returns the maximum depth of the tree -- the number of nodes along the longest path from this node down to the farthest leaf node.

deviance

double deviance()

Returns the deviance of node.

merge

Node merge()

Try to merge the children nodes and return a leaf node. If not able to merge, return this node itself.

toString

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.

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.