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
Trainsmodels
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 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, VERBOSE_DEFAULT, VERBOSE_PARAM
-
Fields inherited from interface opennlp.tools.ml.EventTrainer
EVENT_VALUE
-
-
Constructor Summary
Constructors Constructor Description NaiveBayesTrainer()
Instantiates aNaiveBayesTrainer
with default training parameters.NaiveBayesTrainer(TrainingParameters parameters)
Instantiates aNaiveBayesTrainer
with specificTrainingParameters
.
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description AbstractModel
doTrain(DataIndexer indexer)
boolean
isSortAndMerge()
AbstractModel
trainModel(DataIndexer di)
Trains aNaiveBayesModel
with given parameters.-
Methods inherited from class opennlp.tools.ml.AbstractEventTrainer
getDataIndexer, isValid, train, train, validate
-
Methods inherited from class opennlp.tools.ml.AbstractTrainer
getAlgorithm, getCutoff, getIterations, init, init
-
-
-
-
Field Detail
-
NAIVE_BAYES_VALUE
public static final String NAIVE_BAYES_VALUE
- See Also:
- Constant Field Values
-
-
Constructor Detail
-
NaiveBayesTrainer
public NaiveBayesTrainer()
Instantiates aNaiveBayesTrainer
with default training parameters.
-
NaiveBayesTrainer
public NaiveBayesTrainer(TrainingParameters parameters)
Instantiates aNaiveBayesTrainer
with specificTrainingParameters
.- Parameters:
parameters
- Theparameter
to use.
-
-
Method Detail
-
isSortAndMerge
public boolean isSortAndMerge()
- Specified by:
isSortAndMerge
in classAbstractEventTrainer
-
doTrain
public AbstractModel doTrain(DataIndexer indexer) throws IOException
- Specified by:
doTrain
in classAbstractEventTrainer
- Throws:
IOException
-
trainModel
public AbstractModel trainModel(DataIndexer di)
Trains aNaiveBayesModel
with given parameters.- Parameters:
di
- TheDataIndexer
used as data input.- Returns:
- A valid, trained
Naive Bayes model
.
-
-