train.py 文件源码

def evaluate(args, model, data):
    train_predict = model.predict(data.trainX)
    print("TRAINING RESULTS")
    print(classification_report(
        [e[1] for e in data.trainY],
        [utils.get_sentiment(e[1]) for e in train_predict],
    ))
    print()

    test_predict = model.predict(data.valX)
    print("DEV RESULTS")
    print(classification_report(
        [e[1] for e in data.valY],
        [utils.get_sentiment(e[1]) for e in test_predict],
    ))
    print()

    if args['--evaluate-test']:
        test_predict = model.predict(data.testX)
        print("TEST RESULTS")
        print(classification_report(
            [e[1] for e in data.testY],
            [utils.get_sentiment(e[1]) for e in test_predict],
        ))
        print()