diff --git a/fluid/PaddleRec/gru4rec/README.md b/fluid/PaddleRec/gru4rec/README.md index 9c4a5247fe2ecb64e79ba96e0922f0fd1750aa8e..353781567f7012996199e51169233b306cd18722 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 a5da585ac23e6f65213b358c9d16ed10fb091b05..ec113260bb4391cb970c515a2c958ae71b6176b7 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 0915fe18d571ba459930960d7a39735dc075c930..8839d6ceee2959b81edc2e50ef3b7ae9049c1a1f 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 1b1736cf937723bc86693c0d8cd39e579735f129..2642452024810fe16cfa1154e273febdb1d63254 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(