Add head_model and nominal_mention_model command line arguments.

Bartłomiej Nitoń
1 parent 1ed8f98c
Showing 1 changed file with 12 additions and 4 deletions
src/main/java/pl/waw/ipipan/zil/core/md/Main.java
@@ -144,7 +144,7 @@ public class Main {
         if (args.length != 2 && args.length != 3) {
             logger.error("Wrong usage! should be: " + Main.class.getSimpleName()
-                    + " input_dir result_dir [zero_subject_model]");
+                    + " input_dir result_dir [head_model] [nominal_mention_model] [zero_subject_model]");
             return;
         }
@@ -159,13 +159,21 @@ public class Main {
             logger.error(outputDir + " is not a directory!");
             return;
         }
-        if (args.length == 3) {
+        if (args.length == 5) {
             try {
+            	InputStream headDetectionModelStream;
+                headDetectionModelStream = new FileInputStream(new File(args[3]));
+                headModel = new HeadDetector(headDetectionModelStream);
+            	
+            	InputStream nominalMentionsDetectionModelStream;
+            	nominalMentionsDetectionModelStream = new FileInputStream(new File(args[4]));
+                nominalMentionModel = new NominalMentionDetector(nominalMentionsDetectionModelStream);
+            	
                 InputStream zeroSubjectDetectionModelStream;
-                zeroSubjectDetectionModelStream = new FileInputStream(new File(args[2]));
+                zeroSubjectDetectionModelStream = new FileInputStream(new File(args[5]));
                 zeroSubjectModel = new ZeroSubjectDetector(zeroSubjectDetectionModelStream);
             } catch (IOException e) {
-                logger.error("Unable to load model from file: " + args[2] + ": " + e, e);
+                logger.error("Unable to load model: " + e, e);
                 return;
             }
         }