コード例 #1
0
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))