weka.classifiers.trees.ht

Class NBNodeAdaptive

java.lang.Object

weka.classifiers.trees.ht.HNode

weka.classifiers.trees.ht.LeafNode

weka.classifiers.trees.ht.ActiveHNode

weka.classifiers.trees.ht.NBNode

weka.classifiers.trees.ht.NBNodeAdaptive

All Implemented Interfaces:

java.io.Serializable, LearningNode

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

Implements a LearningNode that chooses between using majority class or naive Bayes for prediction

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected double	m_majClassCorrectWeight The number of correct predictions made by the majority class
protected double	m_nbCorrectWeight The number of correct predictions made by naive Bayes

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

m_bayes, m_nbWeightThreshold

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
NBNodeAdaptive(Instances header, double nbWeightThreshold) Constructor

Method Summary

Modifier and Type	Method and Description
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 java.lang.String	majorityClass()
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.NBNode

bypassNB

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_majClassCorrectWeight

protected double m_majClassCorrectWeight

The number of correct predictions made by the majority class

m_nbCorrectWeight

protected double m_nbCorrectWeight

The number of correct predictions made by naive Bayes

Constructor Detail

NBNodeAdaptive

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

Constructor

Parameters:

header - the structure of the instances we're training from

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

Throws:

java.lang.Exception - if a problem occurs

Method Detail

majorityClass

protected java.lang.String majorityClass()

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 NBNode

Parameters:

inst - the instance to update with

Throws:

java.lang.Exception - if a problem occurs

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 NBNode

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 NBNode

printLeafModels

protected void printLeafModels(java.lang.StringBuffer buff)

Overrides:

printLeafModels in class NBNode