diff --git a/interface/command_line_tool.py b/interface/command_line_tool.py index e491df7..462aec1 100755 --- a/interface/command_line_tool.py +++ b/interface/command_line_tool.py @@ -229,15 +229,13 @@ def train(self): if args.pnnl: dt_list.append("PNL") - name_test_dataset.append("PNL") - + if args.mclean: dt_list.append("McLean") - name_test_dataset.append("McLean") - + if args.cbm: dt_list.append("CBM") - name_test_dataset.append("CBM") + logging.debug("Number of training dataset: {}".format(len(training_datasets))) logging.debug("Training dataset list: {}".format(dt_list))