From 6eccaa8253a85ac2f3dbee5673022f18c513fe1f Mon Sep 17 00:00:00 2001 From: zhangwenhui03 Date: Wed, 13 Mar 2019 17:51:43 +0800 Subject: [PATCH] fix gru4rec style --- fluid/PaddleRec/gru4rec/README.md | 4 ++-- fluid/PaddleRec/gru4rec/infer.py | 3 ++- fluid/PaddleRec/gru4rec/infer_sample_neg.py | 6 +++--- fluid/PaddleRec/gru4rec/train_sample_neg.py | 2 +- 4 files changed, 8 insertions(+), 7 deletions(-) diff --git a/fluid/PaddleRec/gru4rec/README.md b/fluid/PaddleRec/gru4rec/README.md index 9c4a5247..35378156 100644 --- a/fluid/PaddleRec/gru4rec/README.md +++ b/fluid/PaddleRec/gru4rec/README.md @@ -249,10 +249,10 @@ model saved in model_recall20/epoch_1 ``` ## 预测 -运行命令 开始预测. +运行命令 全词表运行infer.py, 负采样运行infer_sample_neg.py。 ``` -CUDA_VISIBLE_DEVICES=0 python infer.py --test_dir test_data/ --model_dir model_recall20/ --start_index 1 --last_index 10 --use_cuda 1 +CUDA_VISIBLE_DEVICES=0 python infer.py --test_dir test_data/ --model_dir model_output/ --start_index 1 --last_index 10 --use_cuda 1 ``` ## 预测结果示例 diff --git a/fluid/PaddleRec/gru4rec/infer.py b/fluid/PaddleRec/gru4rec/infer.py index a5da585a..ec113260 100644 --- a/fluid/PaddleRec/gru4rec/infer.py +++ b/fluid/PaddleRec/gru4rec/infer.py @@ -63,7 +63,8 @@ def infer(test_reader, use_cuda, model_path): accum_num_sum += (data_length) accum_num_recall += (data_length * acc_) if step_id % 1 == 0: - print("step:%d " % (step_id), accum_num_recall / accum_num_sum) + print("step:%d recall@20:%.4f" % + (step_id, accum_num_recall / accum_num_sum)) t1 = time.time() print("model:%s recall@20:%.3f time_cost(s):%.2f" % (model_path, accum_num_recall / accum_num_sum, t1 - t0)) diff --git a/fluid/PaddleRec/gru4rec/infer_sample_neg.py b/fluid/PaddleRec/gru4rec/infer_sample_neg.py index 0915fe18..8839d6ce 100644 --- a/fluid/PaddleRec/gru4rec/infer_sample_neg.py +++ b/fluid/PaddleRec/gru4rec/infer_sample_neg.py @@ -21,7 +21,7 @@ def parse_args(): parser.add_argument( '--last_index', type=int, default='3', help='last index') parser.add_argument( - '--model_dir', type=str, default='model_bpr_recall20', help='model dir') + '--model_dir', type=str, default='model_neg_recall20', help='model dir') parser.add_argument( '--use_cuda', type=int, default='0', help='whether use cuda') parser.add_argument( @@ -76,8 +76,8 @@ def infer(args, vocab_size, test_reader, use_cuda): accum_num_sum += (data_length) accum_num_recall += (data_length * acc_) if step_id % 1 == 0: - print("step:%d " % (step_id), - accum_num_recall / accum_num_sum) + print("step:%d recall@20:%.4f" % + (step_id, accum_num_recall / accum_num_sum)) t1 = time.time() print("model:%s recall@20:%.4f time_cost(s):%.2f" % (model_path, accum_num_recall / accum_num_sum, t1 - t0)) diff --git a/fluid/PaddleRec/gru4rec/train_sample_neg.py b/fluid/PaddleRec/gru4rec/train_sample_neg.py index 1b1736cf..26424520 100644 --- a/fluid/PaddleRec/gru4rec/train_sample_neg.py +++ b/fluid/PaddleRec/gru4rec/train_sample_neg.py @@ -29,7 +29,7 @@ def parse_args(): parser.add_argument( '--loss', type=str, default="bpr", help='loss: bpr/cross_entropy') parser.add_argument( - '--model_dir', type=str, default='model_bpr_recall20', help='model dir') + '--model_dir', type=str, default='model_neg_recall20', help='model dir') parser.add_argument( '--batch_size', type=int, default=5, help='num of batch size') parser.add_argument( -- GitLab