Skip to content

Commit 7cfe354

Browse files
authored
Merge pull request #234 from 123malin/model_fix
w2v infer bug
2 parents 8fd5aed + 060d7e4 commit 7cfe354

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

models/recall/word2vec/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -242,7 +242,7 @@ python -m paddlerec.run -m /home/your/dir/config.yaml #调试模式 直接指定
242242
243243
2. 使用自定义预测程序预测全量测试集:
244244
```
245-
python infer.py --test_dir ./data/all_test --dict_path ./data/all_dict/word_id_dict.txt --batch_size 20000 --model_dir ./increment_w2v/ --start_index 0 --last_index 5 --emb_size 300
245+
python infer.py --test_dir ./data/all_test --dict_path ./data/all_dict/word_id_dict.txt --batch_size 10000 --model_dir ./increment_w2v/ --start_index 0 --last_index 4 --emb_size 300
246246
```
247247
248248
结论:使用cpu训练5轮,自定义预测准确率为0.540,每轮训练时间7小时左右。

0 commit comments

Comments
 (0)