Package weka.classifiers.trees.j48
Class NBTreeSplit
- java.lang.Object
-
- weka.classifiers.trees.j48.ClassifierSplitModel
-
- weka.classifiers.trees.j48.NBTreeSplit
-
- All Implemented Interfaces:
java.io.Serializable
,java.lang.Cloneable
,RevisionHandler
public class NBTreeSplit extends ClassifierSplitModel
Class implementing a NBTree split on an attribute.- Version:
- $Revision: 1.5 $
- Author:
- Mark Hall (mhall@cs.waikato.ac.nz)
- See Also:
- Serialized Form
-
-
Constructor Summary
Constructors Constructor Description NBTreeSplit(int attIndex, int minNoObj, double sumOfWeights)
Initializes the split model.
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description int
attIndex()
Returns index of attribute for which split was generated.void
buildClassifier(Instances trainInstances)
Creates a NBTree-type split on the given data.double
classProb(int classIndex, Instance instance, int theSubset)
Return the probability for a class valuedouble
getErrors()
Return the errors made by the naive bayes models arising from this split.NBTreeNoSplit
getGlobalModel()
Return the global naive bayes model for this nodejava.lang.String
getRevision()
Returns the revision string.java.lang.String
leftSide(Instances data)
Prints left side of condition..java.lang.String
rightSide(int index, Instances data)
Prints the condition satisfied by instances in a subset.void
setGlobalModel(NBTreeNoSplit global)
Set the global naive bayes model for this nodejava.lang.String
sourceExpression(int index, Instances data)
Returns a string containing java source code equivalent to the test made at this node.double[]
weights(Instance instance)
Returns weights if instance is assigned to more than one subset.int
whichSubset(Instance instance)
Returns index of subset instance is assigned to.-
Methods inherited from class weka.classifiers.trees.j48.ClassifierSplitModel
checkModel, classifyInstance, classProbLaplace, clone, codingCost, distribution, dumpLabel, dumpModel, numSubsets, resetDistribution, sourceClass, split
-
-
-
-
Method Detail
-
buildClassifier
public void buildClassifier(Instances trainInstances) throws java.lang.Exception
Creates a NBTree-type split on the given data. Assumes that none of the class values is missing.- Specified by:
buildClassifier
in classClassifierSplitModel
- Throws:
java.lang.Exception
- if something goes wrong
-
attIndex
public final int attIndex()
Returns index of attribute for which split was generated.
-
whichSubset
public final int whichSubset(Instance instance) throws java.lang.Exception
Returns index of subset instance is assigned to. Returns -1 if instance is assigned to more than one subset.- Specified by:
whichSubset
in classClassifierSplitModel
- Throws:
java.lang.Exception
- if something goes wrong
-
weights
public final double[] weights(Instance instance)
Returns weights if instance is assigned to more than one subset. Returns null if instance is only assigned to one subset.- Specified by:
weights
in classClassifierSplitModel
-
sourceExpression
public final java.lang.String sourceExpression(int index, Instances data)
Returns a string containing java source code equivalent to the test made at this node. The instance being tested is called "i".- Specified by:
sourceExpression
in classClassifierSplitModel
- Parameters:
index
- index of the nominal value testeddata
- the data containing instance structure info- Returns:
- a value of type 'String'
-
rightSide
public final java.lang.String rightSide(int index, Instances data)
Prints the condition satisfied by instances in a subset.- Specified by:
rightSide
in classClassifierSplitModel
- Parameters:
index
- of subsetdata
- training set.
-
leftSide
public final java.lang.String leftSide(Instances data)
Prints left side of condition..- Specified by:
leftSide
in classClassifierSplitModel
- Parameters:
data
- training set.
-
classProb
public double classProb(int classIndex, Instance instance, int theSubset) throws java.lang.Exception
Return the probability for a class value- Overrides:
classProb
in classClassifierSplitModel
- Parameters:
classIndex
- the index of the class valueinstance
- the instance to generate a probability fortheSubset
- the subset to consider- Returns:
- a probability
- Throws:
java.lang.Exception
- if an error occurs
-
getGlobalModel
public NBTreeNoSplit getGlobalModel()
Return the global naive bayes model for this node- Returns:
- a
NBTreeNoSplit
value
-
setGlobalModel
public void setGlobalModel(NBTreeNoSplit global)
Set the global naive bayes model for this node- Parameters:
global
- aNBTreeNoSplit
value
-
getErrors
public double getErrors()
Return the errors made by the naive bayes models arising from this split.- Returns:
- a
double
value
-
getRevision
public java.lang.String getRevision()
Returns the revision string.- Returns:
- the revision
-
-