commit 8378164402496300714b6078f4d15696b344632e
author Mark Giaconia <markg@apache.org> Sun Feb 02 14:39:41 2014 +0000
committer Mark Giaconia <markg@apache.org> Sun Feb 02 14:39:41 2014 +0000
tree 3f07847ac14453bcbc82d8fceb213b48948ffb25
parent b9f636191d314792e173ec193d34a1d6a9918c6d

OPENNLP-600
Changed to MockInputStreamFactory for training

modelbuilder-addon/src/main/java/opennlp/addons/modelbuilder/impls/GenericModelableImpl.java

diff --git a/modelbuilder-addon/src/main/java/opennlp/addons/modelbuilder/impls/GenericModelableImpl.java b/modelbuilder-addon/src/main/java/opennlp/addons/modelbuilder/impls/GenericModelableImpl.java
index 572e84b..815d548 100644
--- a/modelbuilder-addon/src/main/java/opennlp/addons/modelbuilder/impls/GenericModelableImpl.java
+++ b/modelbuilder-addon/src/main/java/opennlp/addons/modelbuilder/impls/GenericModelableImpl.java
@@ -31,6 +31,7 @@
 import opennlp.tools.namefind.NameSample;

 import opennlp.tools.namefind.NameSampleDataStream;

 import opennlp.tools.namefind.TokenNameFinderModel;

+import opennlp.tools.util.MockInputStreamFactory;

 import opennlp.tools.util.ObjectStream;

 import opennlp.tools.util.PlainTextByLineStream;

 

@@ -90,7 +91,7 @@
       System.out.println("\t\treading training data...");

       Charset charset = Charset.forName("UTF-8");

       ObjectStream<String> lineStream =

-              new PlainTextByLineStream(new FileInputStream(params.getAnnotatedTrainingDataFile()), charset);

+              new PlainTextByLineStream(new MockInputStreamFactory(new FileInputStream(params.getAnnotatedTrainingDataFile())), charset);

       ObjectStream<NameSample> sampleStream = new NameSampleDataStream(lineStream);

 

       TokenNameFinderModel model;