From 8b492eb07bb7064e3529737e4170adf609932dd6 Mon Sep 17 00:00:00 2001 From: bniton <bartek.niton@gmail.com> Date: Mon, 5 Nov 2018 15:43:44 +0100 Subject: [PATCH] Add head_model and nominal_mention_model command line arguments. --- src/main/java/pl/waw/ipipan/zil/core/md/Main.java | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/src/main/java/pl/waw/ipipan/zil/core/md/Main.java b/src/main/java/pl/waw/ipipan/zil/core/md/Main.java index d41fff9..eea75e5 100644 --- a/src/main/java/pl/waw/ipipan/zil/core/md/Main.java +++ b/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; } } -- libgit2 0.22.2