com.sjm.machlearn.classifiers.naivebayes
Class NaiveBayes

java.lang.Object
  |
  +--com.sjm.machlearn.util.MainClass
        |
        +--com.sjm.machlearn.classifiers.Classifier
              |
              +--com.sjm.machlearn.classifiers.naivebayes.NaiveBayes

public class NaiveBayes

extends Classifier

Field Summary

protected Cont2DiscConverter	converter
protected FeatureCounts[][]	featureProbs
protected int	nbins
protected FeatureCounts	outputProbs
protected DataSet	trainingData

Fields inherited from class com.sjm.machlearn.classifiers.Classifier

Fold10Validation, JackKnifeValidation, Random10Validation

Fields inherited from class com.sjm.machlearn.util.MainClass

debug, debug_level, debug_listeners

Constructor Summary

NaiveBayes()

NaiveBayes(int nb)

Method Summary

Feature	classify(Example example)
Classifier	cloneClassifier()
protected void	initializeFeatures()
java.lang.String	printClassifier()
void	setNumBins(int nb)
void	setParameter(int i, java.lang.Object par)
void	train(DataSet td)
void	tune(DataSet trainingData, java.lang.Object[] parameters)

Methods inherited from class com.sjm.machlearn.classifiers.Classifier

doTune, generateROCCurve, getAccuracy, getBothCat, getExampleWeight, getMisCat

Methods inherited from class com.sjm.machlearn.util.MainClass

_internalError, _internalError, addDebugListener, debugMesg, debugMesg, debugMesg, debugMesg, debugMesg, debugMesg, debugOff, debugOn, internalError, internalError, setDebug, setDebugLevel

Methods inherited from class java.lang.Object

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

Field Detail

outputProbs

protected FeatureCounts outputProbs

featureProbs

protected FeatureCounts[][] featureProbs

trainingData

protected DataSet trainingData

nbins

protected int nbins

converter

protected Cont2DiscConverter converter

Constructor Detail

NaiveBayes

public NaiveBayes()

NaiveBayes

public NaiveBayes(int nb)

Method Detail

setNumBins

public void setNumBins(int nb)

initializeFeatures

protected void initializeFeatures()

train

public void train(DataSet td)

Overrides:

train in class Classifier

classify

public Feature classify(Example example)

Overrides:

classify in class Classifier

tune

public void tune(DataSet trainingData,
                 java.lang.Object[] parameters)

printClassifier

public java.lang.String printClassifier()

Overrides:

printClassifier in class Classifier

cloneClassifier

public Classifier cloneClassifier()

Overrides:

cloneClassifier in class Classifier

setParameter

public void setParameter(int i,
                         java.lang.Object par)

Overrides:

setParameter in class Classifier