OPENNLP-615
Fixed doccat training portion due to recent changes using inputStreamFactory
diff --git a/geoentitylinker-addon/src/main/java/opennlp/addons/geoentitylinker/GeoEntityLinkerSetupUtils.java b/geoentitylinker-addon/src/main/java/opennlp/addons/geoentitylinker/GeoEntityLinkerSetupUtils.java
index 16b2d12..de8af7b 100644
--- a/geoentitylinker-addon/src/main/java/opennlp/addons/geoentitylinker/GeoEntityLinkerSetupUtils.java
+++ b/geoentitylinker-addon/src/main/java/opennlp/addons/geoentitylinker/GeoEntityLinkerSetupUtils.java
@@ -39,7 +39,7 @@
import opennlp.tools.cmdline.MarkableFileInputStreamFactory;
import opennlp.tools.ml.model.DataIndexer;
import opennlp.tools.util.InputStreamFactory;
-import opennlp.tools.util.MockInputStreamFactory;
+
/**
*
@@ -112,7 +112,7 @@
try {
- ObjectStream<String> lineStream = new PlainTextByLineStream(new MockInputStreamFactory(dataIn), "UTF-8");
+ ObjectStream<String> lineStream = new PlainTextByLineStream(new MarkableFileInputStreamFactory(annotationOutFile), "UTF-8");
ObjectStream<DocumentSample> sampleStream = new DocumentSampleStream(lineStream);
model = DocumentCategorizerME.train("en", sampleStream);
