Package opennlp.tools.ml.maxent

Class GISTrainer

java.lang.Object
opennlp.tools.ml.AbstractTrainer
opennlp.tools.ml.AbstractEventTrainer
opennlp.tools.ml.maxent.GISTrainer
All Implemented Interfaces:
EventTrainer
public class GISTrainer extends AbstractEventTrainer
An implementation of Generalized Iterative Scaling. The reference paper for this implementation was Adwait Ratnaparkhi's tech report at the University of Pennsylvania's Institute for Research in Cognitive Science, and is available at ftp://ftp.cis.upenn.edu/pub/ircs/tr/97-08.ps.Z.

The slack parameter used in the above implementation has been removed by default from the computation and a method for updating with Gaussian smoothing has been added per Investigating GIS and Smoothing for Maximum Entropy Taggers, Clark and Curran (2002). http://acl.ldc.upenn.edu/E/E03/E03-1071.pdf The slack parameter can be used by setting useSlackParameter to true. Gaussian smoothing can be used by setting useGaussianSmoothing to true.

A prior can be used to train models which converge to the distribution which minimizes the relative entropy between the distribution specified by the empirical constraints of the training data and the specified prior. By default, the uniform distribution is used as the prior.

  • Field Details

  • Constructor Details

    • GISTrainer

      public GISTrainer()
      Creates a new GISTrainer instance which does not print progress messages about training to STDOUT.

  • Method Details

    • isSortAndMerge

      public boolean isSortAndMerge()
      Specified by:
      isSortAndMerge in class AbstractEventTrainer

    • init

      public void init(TrainingParameters trainingParameters, Map<String,String> reportMap)
      Specified by:
      init in interface EventTrainer
      Overrides:
      init in class AbstractTrainer

    • doTrain

      public MaxentModel doTrain(DataIndexer indexer) throws IOException
      Specified by:
      doTrain in class AbstractEventTrainer
      Throws:
      IOException

    • setSmoothing

      public void setSmoothing(boolean smooth)
      Sets whether this trainer will use smoothing while training the model. This can improve model accuracy, though training will potentially take longer and use more memory. Model size will also be larger.
      Parameters:
      smooth - true if smoothing is desired, false if not

    • setSmoothingObservation

      public void setSmoothingObservation(double timesSeen)
      Sets whether this trainer will use smoothing while training the model. This can improve model accuracy, though training will potentially take longer and use more memory. Model size will also be larger.
      Parameters:
      timesSeen - the "number" of times we want the trainer to imagine it saw a feature that it actually didn't see

    • setGaussianSigma

      public void setGaussianSigma(double sigmaValue)
      Sets whether this trainer will use smoothing while training the model. This can improve model accuracy, though training will potentially take longer and use more memory. Model size will also be larger.

    • trainModel

      public GISModel trainModel(ObjectStream<Event> eventStream) throws IOException
      Train a model using the GIS algorithm, assuming 100 iterations and no cutoff.
      Parameters:
      eventStream - The EventStream holding the data on which this model will be trained.
      Returns:
      The newly trained model, which can be used immediately or saved to disk using an opennlp.tools.ml.maxent.io.GISModelWriter object.
      Throws:
      IOException

    • trainModel

      public GISModel trainModel(ObjectStream<Event> eventStream, int iterations, int cutoff) throws IOException
      Trains a GIS model on the event in the specified event stream, using the specified number of iterations and the specified count cutoff.
      Parameters:
      eventStream - A stream of all events.
      iterations - The number of iterations to use for GIS.
      cutoff - The number of times a feature must occur to be included.
      Returns:
      A GIS model trained with specified
      Throws:
      IOException

    • trainModel

      public GISModel trainModel(int iterations, DataIndexer di)
      Train a model using the GIS algorithm.
      Parameters:
      iterations - The number of GIS iterations to perform.
      di - The data indexer used to compress events in memory.
      Returns:
      The newly trained model, which can be used immediately or saved to disk using an opennlp.tools.ml.maxent.io.GISModelWriter object.

    • trainModel

      public GISModel trainModel(int iterations, DataIndexer di, int threads)
      Train a model using the GIS algorithm.
      Parameters:
      iterations - The number of GIS iterations to perform.
      di - The data indexer used to compress events in memory.
      threads -
      Returns:
      The newly trained model, which can be used immediately or saved to disk using an opennlp.tools.ml.maxent.io.GISModelWriter object.

    • trainModel

      public GISModel trainModel(int iterations, DataIndexer di, Prior modelPrior, int threads)
      Train a model using the GIS algorithm.
      Parameters:
      iterations - The number of GIS iterations to perform.
      di - The data indexer used to compress events in memory.
      modelPrior - The prior distribution used to train this model.
      Returns:
      The newly trained model, which can be used immediately or saved to disk using an opennlp.tools.ml.maxent.io.GISModelWriter object.