def run(sysargs): if len(sysargs) < 1: print("Insufficient input args.") print("Usage:") print("python lstm.py <input_file_path>") else: skip_train_flag = False testbed = TestBed() if (len(sysargs) == 2): skip_train_flag = literal_eval(sysargs[1]) print("\nskip_train_flag:'" + str(skip_train_flag) + "'") input_file_path = sysargs[0] dr = DataReader() dr.read_pkl_data_at_file_path(input_file_path) sequences = dr.get_sequences() labels = dr.get_labels() if not skip_train_flag: # train testbed.init_model() testbed.train(sequences, labels) testbed.save_model() else: # skipping training part, load model testbed.load_model() metrics_names, score = testbed.test(sequences, labels) print("metrics_names:") print(metrics_names) print("score=" + str(score))