keel.Algorithms.Decision_Trees.CART.tree

Class TreeNode

java.lang.Object

keel.Algorithms.Decision_Trees.CART.tree.TreeNode

public class TreeNode
extends java.lang.Object

Class that implements a tree used by the CART algorithm

Constructor Summary

Constructors

Constructor and Description
TreeNode(TreeNode parent) Default Constructor.
TreeNode(TreeNode parent, int[] patterns) Constructor.

Method Summary

Modifier and Type	Method and Description
int	depth()
double	evaluate(double[] pattern, boolean regression) Given a instance, returns its class or regression value.
double	getImpurities()
TreeNode	getLeftSon() It gets the left son of the current node
int	getOutputClass()
double	getOutputValue()
TreeNode	getParent() It gets the parent of the current node
int[]	getPatterns()
TreeNode	getRightSon()
double	getValue()
int	getVariable()
boolean	isTerminal()
void	setImpurities(double impurities)
void	setLeftSon(TreeNode leftSon)
void	setOutputClass(int outputClass)
void	setOutputValue(double outputValue)
void	setParent(TreeNode parent) It sets the parent of the current node
void	setPatterns(int[] patterns)
void	setRightSon(TreeNode rightSon)
void	setValue(double value)
void	setVariable(int variable)
java.lang.String	toString() Returns a String representation of the node and its descendents.

Methods inherited from class java.lang.Object

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

Constructor Detail

TreeNode

public TreeNode(TreeNode parent)

Default Constructor. Sets the tree parent node.

Parameters:

parent - parent node to be set to the new node.

TreeNode

public TreeNode(TreeNode parent,
                int[] patterns)

Constructor. Sets the tree parent node and patterns.

Parameters:

parent - parent node to be set to the new node.

patterns - Index of patterns in this node

Method Detail

getParent

public TreeNode getParent()

It gets the parent of the current node

Returns:

parent parent of current node

setParent

public void setParent(TreeNode parent)

It sets the parent of the current node

Parameters:

parent - to set

getLeftSon

public TreeNode getLeftSon()

It gets the left son of the current node

Returns:

left son of current node. null if it is terminal

setLeftSon

public void setLeftSon(TreeNode leftSon)

Parameters:

leftSon - son to set

getRightSon

public TreeNode getRightSon()

Returns:

right son of current node. null if it is terminal

setRightSon

public void setRightSon(TreeNode rightSon)

Parameters:

rightSon - to set

getPatterns

public int[] getPatterns()

Returns:

the patterns

setPatterns

public void setPatterns(int[] patterns)

Parameters:

patterns - the patterns to set

getVariable

public int getVariable()

Returns:

the variable

setVariable

public void setVariable(int variable)

Parameters:

variable - the variable to set

getValue

public double getValue()

Returns:

the value

setValue

public void setValue(double value)

Parameters:

value - the value to set

getOutputClass

public int getOutputClass()

Returns:

the outputClass

setOutputClass

public void setOutputClass(int outputClass)

Parameters:

outputClass - the outputClass to set

getOutputValue

public double getOutputValue()

Returns:

the outputValue

setOutputValue

public void setOutputValue(double outputValue)

Parameters:

outputValue - the outputValue to set

getImpurities

public double getImpurities()

Returns:

the impurities

setImpurities

public void setImpurities(double impurities)

Parameters:

impurities - the impurities to set

depth

public int depth()

Returns:

depth of current node

isTerminal

public boolean isTerminal()

Returns:

true if node is leaf; false otherwise

evaluate

public double evaluate(double[] pattern,
                       boolean regression)

Given a instance, returns its class or regression value.

Parameters:

pattern - pattern to evaluate

regression - flag to determine whether it is a regression or classification problem

Returns:

class or regression value.

toString

public java.lang.String toString()

Returns a String representation of the node and its descendents.

Overrides:

toString in class java.lang.Object

Returns:

String representation of the node and its descendents.