OPENNLP-600
Changed to MockInputStreamFactory for doccat model training
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 06581a4..16b2d12 100644
--- a/geoentitylinker-addon/src/main/java/opennlp/addons/geoentitylinker/GeoEntityLinkerSetupUtils.java
+++ b/geoentitylinker-addon/src/main/java/opennlp/addons/geoentitylinker/GeoEntityLinkerSetupUtils.java
@@ -36,6 +36,10 @@
import opennlp.tools.util.ObjectStream;
import opennlp.tools.util.PlainTextByLineStream;
import static opennlp.addons.geoentitylinker.ModelBasedScorer.RADIUS;
+import opennlp.tools.cmdline.MarkableFileInputStreamFactory;
+import opennlp.tools.ml.model.DataIndexer;
+import opennlp.tools.util.InputStreamFactory;
+import opennlp.tools.util.MockInputStreamFactory;
/**
*
@@ -107,8 +111,8 @@
InputStream dataIn = new FileInputStream(annotationOutFile);
try {
- ObjectStream<String> lineStream =
- new PlainTextByLineStream(dataIn, "UTF-8");
+
+ ObjectStream<String> lineStream = new PlainTextByLineStream(new MockInputStreamFactory(dataIn), "UTF-8");
ObjectStream<DocumentSample> sampleStream = new DocumentSampleStream(lineStream);
model = DocumentCategorizerME.train("en", sampleStream);
