Example usage for opennlp.tools.ml BeamSearch BEAM_SIZE_PARAMETER

List of usage examples for opennlp.tools.ml BeamSearch BEAM_SIZE_PARAMETER

Introduction

In this page you can find the example usage for opennlp.tools.ml BeamSearch BEAM_SIZE_PARAMETER.

Prototype

String BEAM_SIZE_PARAMETER

To view the source code for opennlp.tools.ml BeamSearch BEAM_SIZE_PARAMETER.

Click Source Link

Usage

From source file:de.tudarmstadt.ukp.dkpro.core.opennlp.OpenNlpNamedEntityRecognizerTrainer.java

@Override
public void initialize(UimaContext aContext) throws ResourceInitializationException {
    super.initialize(aContext);

    stream = new CasNameSampleStream();

    TrainingParameters params = new TrainingParameters();
    params.put(TrainingParameters.ALGORITHM_PARAM, algorithm);
    //        params.put(TrainingParameters.TRAINER_TYPE_PARAM,
    //                TrainerFactory.getTrainerType(params.getSettings()).name());
    params.put(TrainingParameters.ITERATIONS_PARAM, Integer.toString(iterations));
    params.put(TrainingParameters.CUTOFF_PARAM, Integer.toString(cutoff));
    params.put(BeamSearch.BEAM_SIZE_PARAMETER, Integer.toString(beamSize));

    byte featureGenCfg[] = loadFeatureGen(featureGen);

    Callable<TokenNameFinderModel> trainTask = () -> {
        try {//  w w w  .  j  ava 2s.  c  om
            return NameFinderME.train(language, null, stream, params, new TokenNameFinderFactory(featureGenCfg,
                    Collections.<String, Object>emptyMap(), sequenceEncoding.getCodec()));
        } catch (Throwable e) {
            stream.close();
            throw e;
        }
    };

    future = executor.submit(trainTask);
}