if args.do_val: evaluate(exe, test_prog, test_pyreader, [loss.name, accuracy.name, num_seqs.name], "dev") except fluid.core.EOFException: save_path = os.path.join(args.checkpoints, "step_" + str(steps), "checkpoint") fluid.save(train_program, save_path) train_pyreader.reset() break # final eval on dev set if args.do_val: print("Final validation result:") evaluate(exe, test_prog, test_pyreader, [loss.name, accuracy.name, num_seqs.name], "dev") # final eval on test set if args.do_infer: print("Final test result:") infer(exe, infer_prog, infer_pyreader, [probs.name], "infer") if __name__ == "__main__": args = PDConfig() args.build() args.print_arguments() check_cuda(args.use_cuda) main(args)
(task_name, card_num, ce_loss)) print("kpis\ttrain_acc_%s_card%s\t%f" % (task_name, card_num, ce_acc)) # evaluate on test set if not args.do_train and args.do_val: print("Final test result:") evaluate(test_exe, test_prog, test_loader, [loss.name, accuracy.name, num_seqs.name], "test") # infer if args.do_infer: print("Final infer result:") infer(test_exe, test_prog, infer_loader, [probs.name], "infer") def get_cards(): num = 0 cards = os.environ.get('CUDA_VISIBLE_DEVICES', '') if cards != '': num = len(cards.split(",")) return num if __name__ == "__main__": args = PDConfig('config.json') args.build() args.print_arguments() check_cuda(args.use_cuda) check_version() main(args)
data = [] seq_lens = [] for query in texts: wids = utils.query2ids(args.vocab_path, query) wids, seq_len = utils.pad_wid(wids) data.append(wids) seq_lens.append(seq_len) data = np.array(data) seq_lens = np.array(seq_lens) pred = exe.run(infer_program, feed={feed_names[0]: data, feed_names[1]: seq_lens}, fetch_list=fetch_targets, return_numpy=True) for probs in pred[0]: print("%d\t%f\t%f\t%f" % (np.argmax(probs), probs[0], probs[1], probs[2])) if __name__ == "__main__": args = PDConfig(json_file="./config.json") args.build() args.print_arguments() check_cuda(args.use_cuda) if args.do_save_inference_model: do_save_inference_model(args) else: texts = [u"我 讨厌 你 , 哼哼 哼 。 。", u"我 喜欢 你 , 爱 你 哟"] test_inference_model(args, texts)