From 4826eb9d2b464f7841942520fb7b4de23aea4556 Mon Sep 17 00:00:00 2001 From: William D C M SILVA Date: Mon, 23 Jan 2017 19:31:47 -0200 Subject: [PATCH] OPENNLP-955: Fixes docbegin feature generato The docbegin was mapped to PreviousMapFeatureGenerator instead of DocumentBeginFeatureGenerator. Also changed the name of the factory by appending the sufix. --- .../opennlp/tools/util/featuregen/GeneratorFactory.java | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/opennlp-tools/src/main/java/opennlp/tools/util/featuregen/GeneratorFactory.java b/opennlp-tools/src/main/java/opennlp/tools/util/featuregen/GeneratorFactory.java index bccbea23a..fa97f43a6 100644 --- a/opennlp-tools/src/main/java/opennlp/tools/util/featuregen/GeneratorFactory.java +++ b/opennlp-tools/src/main/java/opennlp/tools/util/featuregen/GeneratorFactory.java @@ -259,15 +259,15 @@ static void register(Map factoryMap) { } } - static class DocumentBeginFeatureGenerator implements XmlFeatureGeneratorFactory { + static class DocumentBeginFeatureGeneratorFactory implements XmlFeatureGeneratorFactory { public AdaptiveFeatureGenerator create(Element generatorElement, FeatureGeneratorResourceProvider resourceManager) { - return new PreviousMapFeatureGenerator(); + return new DocumentBeginFeatureGenerator(); } static void register(Map factoryMap) { - factoryMap.put("docbegin", new DocumentBeginFeatureGenerator()); + factoryMap.put("docbegin", new DocumentBeginFeatureGeneratorFactory()); } } @@ -633,7 +633,7 @@ static void register(Map factoryMap) { CharacterNgramFeatureGeneratorFactory.register(factories); DefinitionFeatureGeneratorFactory.register(factories); DictionaryFeatureGeneratorFactory.register(factories); - DocumentBeginFeatureGenerator.register(factories); + DocumentBeginFeatureGeneratorFactory.register(factories); PreviousMapFeatureGeneratorFactory.register(factories); SentenceFeatureGeneratorFactory.register(factories); TokenClassFeatureGeneratorFactory.register(factories);