cc.mallet.classify
Class NaiveBayesEMTrainer
java.lang.Object
cc.mallet.classify.ClassifierTrainer<NaiveBayes>
cc.mallet.classify.NaiveBayesEMTrainer
public class NaiveBayesEMTrainer
- extends ClassifierTrainer<NaiveBayes>
- Author:
- Andrew McCallum mccallum@cs.umass.edu
Methods inherited from class java.lang.Object |
clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait |
NaiveBayesEMTrainer
public NaiveBayesEMTrainer()
getFeatureMultinomialEstimator
public Multinomial.Estimator getFeatureMultinomialEstimator()
setFeatureMultinomialEstimator
public void setFeatureMultinomialEstimator(Multinomial.Estimator me)
getPriorMultinomialEstimator
public Multinomial.Estimator getPriorMultinomialEstimator()
setPriorMultinomialEstimator
public void setPriorMultinomialEstimator(Multinomial.Estimator me)
setDocLengthNormalization
public void setDocLengthNormalization(double d)
getDocLengthNormalization
public double getDocLengthNormalization()
getUnlabeledDataWeight
public double getUnlabeledDataWeight()
setUnlabeledDataWeight
public void setUnlabeledDataWeight(double unlabeledDataWeight)
getIteration
public int getIteration()
isFinishedTraining
public boolean isFinishedTraining()
- Overrides:
isFinishedTraining
in class ClassifierTrainer<NaiveBayes>
getClassifier
public NaiveBayes getClassifier()
- Specified by:
getClassifier
in class ClassifierTrainer<NaiveBayes>
train
public NaiveBayes train(InstanceList trainingSet)
- Specified by:
train
in class ClassifierTrainer<NaiveBayes>
toString
public java.lang.String toString()
- Overrides:
toString
in class java.lang.Object