From 870f3d84b89c24d6e4ceed25c518bafcfb8ff287 Mon Sep 17 00:00:00 2001 From: malin10 Date: Wed, 23 Sep 2020 12:18:17 +0800 Subject: [PATCH] test=develop, update readme --- models/recall/gru4rec/README.md | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/models/recall/gru4rec/README.md b/models/recall/gru4rec/README.md index c9696b00..a46a2129 100644 --- a/models/recall/gru4rec/README.md +++ b/models/recall/gru4rec/README.md @@ -4,10 +4,10 @@ ``` ├── data #样例数据及数据处理相关文件 - ├── train - ├── small_train.txt # 样例训练数据 + ├── train + ├── small_train.txt # 样例训练数据 ├── test - ├── small_test.txt # 样例测试数据 + ├── small_test.txt # 样例测试数据 ├── convert_format.py # 数据转换脚本 ├── download.py # 数据下载脚本 ├── preprocess.py # 数据预处理脚本 @@ -178,6 +178,7 @@ PaddleRec Finish 用原论文的完整数据复现论文效果需要在config.yaml修改超参: - batch_size: 修改config.yaml中dataset_train数据集的batch_size为500。 - epochs: 修改config.yaml中runner的epochs为10。 +- 数据源:修改config.yaml中dataset_train数据集的data_path为"{workspace}/data/all_train",dataset_test数据集的data_path为"{workspace}/data/all_test"。 使用gpu训练10轮 测试结果为 -- GitLab