Package opennlp.tools.ml.maxent

Class 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.

    • Constructor Detail

      • GISTrainer

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

    • Method Detail

      • 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.