weka.classifiers.trees.ht

Class NBNode

java.lang.Object

weka.classifiers.trees.ht.HNode

weka.classifiers.trees.ht.LeafNode

weka.classifiers.trees.ht.ActiveHNode

weka.classifiers.trees.ht.NBNode

All Implemented Interfaces:

java.io.Serializable, LearningNode

Direct Known Subclasses:

NBNodeAdaptive

public class NBNode
extends ActiveHNode
implements LearningNode, java.io.Serializable

Implements a LearningNode that uses a naive Bayes model

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected NaiveBayesUpdateable	m_bayes The naive Bayes model at the node
protected double	m_nbWeightThreshold The weight of instances that need to be seen by this node before allowing naive Bayes to make predictions

Fields inherited from class weka.classifiers.trees.ht.ActiveHNode

m_nodeStats, m_weightSeenAtLastSplitEval

Fields inherited from class weka.classifiers.trees.ht.LeafNode

m_parentBranch, m_parentNode, m_theNode

Fields inherited from class weka.classifiers.trees.ht.HNode

m_classDistribution, m_leafNum, m_nodeNum

Constructor Summary

Constructors

Constructor and Description
NBNode(Instances header, double nbWeightThreshold) Construct a new NBNode

Method Summary

Modifier and Type	Method and Description
protected double[]	bypassNB(Instance inst, Attribute classAtt)
protected int	dumpTree(int depth, int leafCount, java.lang.StringBuffer buff)
double[]	getDistribution(Instance inst, Attribute classAtt) Return a class probability distribution computed from the frequency counts at this node
protected void	printLeafModels(java.lang.StringBuffer buff)
void	updateNode(Instance inst) Update the node with the supplied instance

Methods inherited from class weka.classifiers.trees.ht.ActiveHNode

getPossibleSplits

Methods inherited from class weka.classifiers.trees.ht.HNode

classDistributionIsPure, graphTree, installNodeNums, isLeaf, leafForInstance, numEntriesInClassDistribution, toString, totalWeight, updateDistribution

Methods inherited from class java.lang.Object

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

Field Detail

m_bayes

protected NaiveBayesUpdateable m_bayes

The naive Bayes model at the node

m_nbWeightThreshold

protected double m_nbWeightThreshold

The weight of instances that need to be seen by this node before allowing naive Bayes to make predictions

Constructor Detail

NBNode

public NBNode(Instances header,
              double nbWeightThreshold)
       throws java.lang.Exception

Construct a new NBNode

Parameters:

header - the instances structure of the data we're learning from

nbWeightThreshold - the weight mass to see before allowing naive Bayes to predict

Throws:

java.lang.Exception - if a problem occurs

Method Detail

updateNode

public void updateNode(Instance inst)
                throws java.lang.Exception

Description copied from class: HNode

Update the node with the supplied instance

Overrides:

updateNode in class ActiveHNode

Parameters:

inst - the instance to update with

Throws:

java.lang.Exception - if a problem occurs

bypassNB

protected double[] bypassNB(Instance inst,
                            Attribute classAtt)
                     throws java.lang.Exception

Throws:

java.lang.Exception

getDistribution

public double[] getDistribution(Instance inst,
                                Attribute classAtt)
                         throws java.lang.Exception

Description copied from class: HNode

Return a class probability distribution computed from the frequency counts at this node

Overrides:

getDistribution in class HNode

Parameters:

inst - the instance to get a prediction for

classAtt - the class attribute

Returns:

a class probability distribution

Throws:

java.lang.Exception - if a problem occurs

dumpTree

protected int dumpTree(int depth,
                       int leafCount,
                       java.lang.StringBuffer buff)

Overrides:

dumpTree in class HNode

printLeafModels

protected void printLeafModels(java.lang.StringBuffer buff)

Overrides:

printLeafModels in class HNode