File tree Expand file tree Collapse file tree 2 files changed +4
-5
lines changed Expand file tree Collapse file tree 2 files changed +4
-5
lines changed Original file line number Diff line number Diff 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```
118118python3 -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
Original file line number Diff line number Diff line change 2323
2424class 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 ):
You can’t perform that action at this time.
0 commit comments