eus.ixa.ixa.pipe.ml

Class DocumentClassifierTrainer

java.lang.Object

eus.ixa.ixa.pipe.ml.DocumentClassifierTrainer

public class DocumentClassifierTrainer
extends Object

Trainer based on Apache OpenNLP Machine Learning API. This class creates a feature set based on the features activated in the docClassicationTrainer.properties file

Version:

2017-05-27

Author:

ragerri

Constructor Summary

Constructors

Constructor and Description
DocumentClassifierTrainer(opennlp.tools.util.TrainingParameters params) Construct a trainer with training and test data and language options.

Method Summary

Modifier and Type	Method and Description
void	createDocumentClassificationFactory(opennlp.tools.util.TrainingParameters params) Create createDocumentClassificationFactory .
DocumentClassifierFactory	getDocumentClassificationFactory() Get the features which are implemented in each of the trainers extending this class.
static opennlp.tools.util.ObjectStream<DocSample>	getDocumentStream(String inputData, String clearFeatures) Getting the stream with the right corpus format.
DocumentClassifierFactory	setDocumentClassifierFactory(DocumentClassifierFactory tokenNameFinderFactory)
DocumentClassifierModel	train(opennlp.tools.util.TrainingParameters params)

Methods inherited from class java.lang.Object

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

Constructor Detail

DocumentClassifierTrainer

public DocumentClassifierTrainer(opennlp.tools.util.TrainingParameters params)
                          throws IOException

Construct a trainer with training and test data and language options.

Parameters:

params - the training parameters

Throws:

IOException - io exception

Method Detail

createDocumentClassificationFactory

public void createDocumentClassificationFactory(opennlp.tools.util.TrainingParameters params)
                                         throws IOException

Create createDocumentClassificationFactory .

Parameters:

params - the parameter training file

Throws:

IOException - if io error

train

public final DocumentClassifierModel train(opennlp.tools.util.TrainingParameters params)

getDocumentStream

public static opennlp.tools.util.ObjectStream<DocSample> getDocumentStream(String inputData,
                                                                           String clearFeatures)
                                                                    throws IOException

Getting the stream with the right corpus format.

Parameters:

inputData - the input data

clearFeatures - whether to reset the features for each document

Returns:

the stream from the several corpus formats

Throws:

IOException - the io exception

getDocumentClassificationFactory

public final DocumentClassifierFactory getDocumentClassificationFactory()

Get the features which are implemented in each of the trainers extending this class.

Returns:

the features

setDocumentClassifierFactory

public final DocumentClassifierFactory setDocumentClassifierFactory(DocumentClassifierFactory tokenNameFinderFactory)