Skip to content

Commit 819c869

Browse files
committed
doc change
1 parent 6870cc0 commit 819c869

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

models/rank/naml/README.md

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ python3 -u ../../../tools/infer.py -m config.yaml
9191
#### Loss及Acc计算
9292
- 预测的结果为一个softmax向量,表示实际浏览文章和负采样文章同时出现的情况下被用户浏览的概率
9393
- 样本的损失函数值由交叉熵给出
94-
- 我们同时还会计算预测的acc,即top1的准确率
94+
- 我们同时还会计算预测的auc
9595

9696
## 效果复现
9797
为了方便使用者能够快速的跑通每一个模型,我们在每个模型下都提供了样例数据。
@@ -111,12 +111,11 @@ python3 -u ../../../tools/trainer.py -m config_bigdata.yaml
111111
以下为训练2个epoch的结果
112112
| 模型 | top1 acc | batch_size | epoch_num| Time of each epoch|
113113
| :------| :------ | :------ | :------| :------ |
114-
| naml | 0.43 | 512 | 4 | 约0.5小时 |
114+
| naml | 0.72 | 50 | 3 | 约4小时 |
115115

116116
预测
117117
```
118118
python3 -u ../../../tools/infer.py -m config_bigdata.yaml
119119
```
120120

121-
期待运行结果如下
122-
INFO - epoch: 1 done, acc: 0.427140, epoch time: 126.27 s
121+
期待预测auc为0.65

models/rank/naml/dygraph_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323

2424
class DygraphModel():
2525
def __init__(self):
26-
self.bucket = 1000000
26+
self.bucket = 100000
2727
self.absolute_limt = 200.0
2828

2929
def rescale(self, number):

0 commit comments

Comments
 (0)