Skip to content

Commit 8fd5aed

Browse files
authored
Merge pull request #231 from 123malin/model_fix
test=develop, bug fix for w2v, youtube_dnn'readme
2 parents a4bd3a6 + f6b622a commit 8fd5aed

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

models/recall/word2vec/README.md

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -222,15 +222,18 @@ Infer phase2 of epoch 3 done, use time: 4.43099021912, global metrics: acc=[1.]
222222
## 论文复现
223223
224224
1. 用原论文的完整数据复现论文效果需要在config.yaml修改超参:
225+
```
225226
- name: dataset_train
226227
batch_size: 100 # 1. 修改batch_size为100
227228
type: DataLoader
228229
data_path: "{workspace}/data/all_train" # 2. 修改数据为全量训练数据
229-
word_count_dict_path: "{workspace}/data/all_dict/ word_count_dict.txt" # 3. 修改词表为全量词表
230+
word_count_dict_path: "{workspace}/data/all_dict/word_count_dict.txt" # 3. 修改词表为全量词表
230231
data_converter: "{workspace}/w2v_reader.py"
232+
- name: dataset_infer
233+
data_path: "{workspace}/data/all_test" # 4. 修改数据为全量测试数据
234+
word_id_dict_path: "{workspace}/data/all_dict/word_id_dict.txt" # 5. 修改词表为全量词表
231235

232-
- name: single_cpu_train
233-
- epochs: # 4. 修改config.yaml中runner的epochs为5。
236+
```
234237
235238
修改后运行方案:修改config.yaml中的'workspace'为config.yaml的目录位置,执行
236239
```

models/recall/youtube_dnn/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
├── data.txt
99
├── test
1010
├── data.txt
11-
├── generate_ramdom_data # 随机训练数据生成文件
11+
├── generate_ramdom_data.py # 随机训练数据生成文件
1212
├── __init__.py
1313
├── README.md # 文档
1414
├── model.py #模型文件
@@ -107,7 +107,7 @@ python infer.py --use_gpu 1 --test_epoch 19 --inference_model_dir ./inference_yo
107107
```
108108
### 运行
109109
```
110-
python -m paddlerec.run -m paddlerec.models.recall.w2v
110+
python -m paddlerec.run -m paddlerec.models.recall.youtube_dnn
111111
```
112112

113113
### 结果展示

0 commit comments

Comments
 (0)