From 48c1e8c86a781d36788892164d997c662084d13c Mon Sep 17 00:00:00 2001 From: Pier-Luc Plante Date: Wed, 17 Apr 2019 09:58:36 -0400 Subject: [PATCH] Fixed the arguments in the train command. Signed-off-by: Pier-Luc Plante --- interface/command_line_tool.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/interface/command_line_tool.py b/interface/command_line_tool.py index 73966cf..1b9cffb 100755 --- a/interface/command_line_tool.py +++ b/interface/command_line_tool.py @@ -133,12 +133,12 @@ def train(self): parser.add_argument("-mtrain", help="Use MetCCS train datasets to create the model", default=False, action="store_true", dest="mtrain") parser.add_argument("-mtest", help="Use MetCCS test datasets to create the model", default=False, - action="store_true", dest="mtestA") + action="store_true", dest="mtest") parser.add_argument("-ast", help="Astarita test datasets to create the model", default=False, - action="store_true", dest="mtestW") + action="store_true", dest="ast") parser.add_argument("-baker", help="Baker dataset to create the model", default=False, action="store_true", - dest="pnnl") + dest="baker") parser.add_argument("-cbm", help="CBM2018 dataset to create the model", default=False, action="store_true", dest="cbm") parser.add_argument("-mclean", help="McLean dataset to create the model", default=False, action="store_true", @@ -163,8 +163,7 @@ def train(self): raise ValueError("Proportion in test set must be between 0 and 1. Recommended: 0.2") logging.debug("\nCondition is: {}".format(not(args.mtrain or args.mtest or args.ast or args.baker or args.cbm or args.mclean))) - if not (args.mtrain or args.mtest or args.ast or args.baker or args.cbm or args.mclean or - args.nd is not None): + if not (args.mtrain or args.mtest or args.ast or args.baker or args.cbm or args.mclean or args.nd is not None): raise ValueError("At least one datafile must be used to train a model.") from DeepCCS.model import DeepCCS