Package opennlp.tools.ml.naivebayes

Class NaiveBayesTrainer

java.lang.Object

opennlp.tools.ml.AbstractTrainer

opennlp.tools.ml.AbstractEventTrainer

opennlp.tools.ml.naivebayes.NaiveBayesTrainer

All Implemented Interfaces:

Trainer, EventTrainer

public class NaiveBayesTrainer
extends AbstractEventTrainer

Trains models using the combination of EM algorithm and Naive Bayes classifier which is described in:

Text Classification from Labeled and Unlabeled Documents using EM Nigam, McCallum, et al. paper of 2000

See Also:

• NaiveBayesModel
• AbstractEventTrainer

Field Summary

Fields

Modifier and Type	Field	Description
static final String	NAIVE_BAYES_VALUE

Fields inherited from class opennlp.tools.ml.AbstractEventTrainer

DATA_INDEXER_ONE_PASS_REAL_VALUE, DATA_INDEXER_ONE_PASS_VALUE, DATA_INDEXER_PARAM, DATA_INDEXER_TWO_PASS_VALUE

Fields inherited from class opennlp.tools.ml.AbstractTrainer

ALGORITHM_PARAM, CUTOFF_DEFAULT, CUTOFF_PARAM, ITERATIONS_DEFAULT, ITERATIONS_PARAM, TRAINER_TYPE_PARAM

Fields inherited from interface opennlp.tools.ml.EventTrainer

EVENT_VALUE

Constructor Summary

Constructors

Constructor	Description
NaiveBayesTrainer()	Instantiates a NaiveBayesTrainer with default training parameters.
NaiveBayesTrainer(TrainingParameters parameters)	Instantiates a NaiveBayesTrainer with specific TrainingParameters.

Method Summary

Modifier and Type	Method	Description
AbstractModel	doTrain(DataIndexer indexer)
boolean	isSortAndMerge()
AbstractModel	trainModel(DataIndexer di)	Trains a NaiveBayesModel with given parameters.

Methods inherited from class opennlp.tools.ml.AbstractEventTrainer

getDataIndexer, train, train, validate

Methods inherited from class opennlp.tools.ml.AbstractTrainer

getAlgorithm, getCutoff, getIterations, init

Methods inherited from class java.lang.Object

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

Methods inherited from interface opennlp.tools.commons.Trainer

init

Field Details

NAIVE_BAYES_VALUE

public static final String NAIVE_BAYES_VALUE

See Also:

• Constant Field Values

Constructor Details

NaiveBayesTrainer

public NaiveBayesTrainer()

Instantiates a NaiveBayesTrainer with default training parameters.

NaiveBayesTrainer

public NaiveBayesTrainer(TrainingParameters parameters)

Instantiates a NaiveBayesTrainer with specific TrainingParameters.

Parameters:

parameters - The parameter to use.

Method Details

isSortAndMerge

public boolean isSortAndMerge()

Specified by:

isSortAndMerge in class AbstractEventTrainer

doTrain

public AbstractModel doTrain(DataIndexer indexer)
                      throws IOException

Specified by:

doTrain in class AbstractEventTrainer

Throws:

IOException

trainModel

public AbstractModel trainModel(DataIndexer di)

Trains a NaiveBayesModel with given parameters.

Parameters:

di - The DataIndexer used as data input.

Returns:

A valid, trained Naive Bayes model.