Skip to content

Commit 4acf869

Browse files
committed
update metrics for mind
1 parent 43ed4cc commit 4acf869

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

models/recall/mind/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -99,8 +99,8 @@ python -u static_infer.py -m config.yaml -top_n 50 #对测试数据进行预测
9999
在全量数据下模型的指标如下:
100100
| 模型 | batch_size | epoch_num| Recall@50 | NDCG@50 | HitRate@50 |Time of each epoch |
101101
| :------| :------ | :------ | :------| :------ | :------| :------ |
102-
| mind(静态图) | 128 | 6 | 5.61% | 8.96% | 11.81% | -- |
103-
| mind(动态图) | 128 | 6 | 5.54% | 8.85% | 11.75% | -- |
102+
| mind | 128 | 20 | 8.43% | 13.28% | 17.22% | -- |
103+
104104

105105
1. 确认您当前所在目录为PaddleRec/models/recall/mind
106106
2. 进入paddlerec/datasets/AmazonBook目录下执行run.sh脚本,会下载处理完成的AmazonBook数据集,并解压到指定目录

0 commit comments

Comments
 (0)