Merge pull request PaddlePaddle#403 from duyiqi17/mind

update metrics for mind
JHSUYU · Mar 26, 2021 · 50ea3df · 50ea3df
2 parents de88a68 + 4acf869
commit 50ea3df
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/recall/mind/README.md b/models/recall/mind/README.md
@@ -99,8 +99,8 @@ python -u static_infer.py -m config.yaml -top_n 50  #对测试数据进行预测
 在全量数据下模型的指标如下：
 | 模型 |  batch_size | epoch_num| Recall@50 | NDCG@50 | HitRate@50 |Time of each epoch |
 | :------| :------ | :------ | :------| :------ | :------|  :------ | 
-| mind(静态图) | 128 | 6 | 5.61% | 8.96% | 11.81% | -- |
-| mind(动态图) | 128 | 6 | 5.54% | 8.85% | 11.75% | -- |
+| mind | 128 | 20 | 8.43% | 13.28% | 17.22% | -- |
+
 
 1. 确认您当前所在目录为PaddleRec/models/recall/mind
 2. 进入paddlerec/datasets/AmazonBook目录下执行run.sh脚本，会下载处理完成的AmazonBook数据集，并解压到指定目录