diff --git a/demo/ernie-classification/cls_predict.py b/demo/ernie-classification/cls_predict.py index 08f538d410bf10b10a969f56cde63d7538ad1da5..6da7889d393df86d02649f8fce6a6f9331999f1d 100644 --- a/demo/ernie-classification/cls_predict.py +++ b/demo/ernie-classification/cls_predict.py @@ -80,4 +80,4 @@ if __name__ == '__main__': for index, batch in enumerate(test_reader()): probs_v = exe.run( feed=data_feeder.feed(batch), fetch_list=[probs.name]) - print(test_examples[index], probs_v[0][0]) + print("%s\tpredict=%s" % (test_examples[index], probs_v[0][0]))