List of usage examples for opennlp.tools.ml BeamSearch BEAM_SIZE_PARAMETER
String BEAM_SIZE_PARAMETER
To view the source code for opennlp.tools.ml BeamSearch BEAM_SIZE_PARAMETER.
Click Source Link
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); }