From ef6d335131bb68e1c81e3623c4702babc5b62c3a Mon Sep 17 00:00:00 2001 From: bniton <bartek.niton@gmail.com> Date: Tue, 6 Nov 2018 13:25:31 +0100 Subject: [PATCH] Fix command line arguments error. --- src/main/java/pl/waw/ipipan/zil/core/md/Main.java | 9 +++++---- 1 file changed, 5 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 eea75e5..b42561d 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 @@ -142,7 +142,7 @@ public class Main { */ public static void main(String[] args) { - if (args.length != 2 && args.length != 3) { + if (args.length != 2 && args.length != 5) { logger.error("Wrong usage! should be: " + Main.class.getSimpleName() + " input_dir result_dir [head_model] [nominal_mention_model] [zero_subject_model]"); return; @@ -159,18 +159,19 @@ public class Main { logger.error(outputDir + " is not a directory!"); return; } + if (args.length == 5) { try { InputStream headDetectionModelStream; - headDetectionModelStream = new FileInputStream(new File(args[3])); + headDetectionModelStream = new FileInputStream(new File(args[2])); headModel = new HeadDetector(headDetectionModelStream); InputStream nominalMentionsDetectionModelStream; - nominalMentionsDetectionModelStream = new FileInputStream(new File(args[4])); + nominalMentionsDetectionModelStream = new FileInputStream(new File(args[3])); nominalMentionModel = new NominalMentionDetector(nominalMentionsDetectionModelStream); InputStream zeroSubjectDetectionModelStream; - zeroSubjectDetectionModelStream = new FileInputStream(new File(args[5])); + zeroSubjectDetectionModelStream = new FileInputStream(new File(args[4])); zeroSubjectModel = new ZeroSubjectDetector(zeroSubjectDetectionModelStream); } catch (IOException e) { logger.error("Unable to load model: " + e, e); -- libgit2 0.22.2