Skip to content

Commit e694b2c

Browse files
authored
Merge pull request #813 from wangzhen38/doc_update
[doc update]update PaddleRec README and dien doc
2 parents 1a6a398 + 962b6b5 commit e694b2c

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

models/rank/dien/README.md

+3-1
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,8 @@ python 3.5/3.6/3.7
7373

7474
os : windows/linux/macos
7575

76+
注:建议通过编译方式安装最新develop分支,不然可能会出现最新属性找不到的错误,安装方式见[教程](https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/develop/install/compile/arm-compile.html)
77+
7678
## 快速开始
7779
本文提供了样例数据可以供您快速体验,在dien模型目录的快速执行命令如下:
7880
```bash
@@ -97,11 +99,11 @@ cat_count: 品类的种类数目
9799
```
98100

99101
#### Loss及Acc计算
100-
- 本文采用AUGRU及模型组网,采用sigmoid激活函数
101102
- 预测的结果为一个sigmoid向量,表示推荐的商品广告被用户点击的概率
102103
- 样本的损失函数值由CTR_loss和Auxiliary_loss叠加组成,同时计算预测的auc
103104

104105
## 效果复现
106+
注:目前paddle未支持AUGRU算子,此处用的是普通的GRU,不同于论文部分。
105107
### 数据集获取及预处理
106108
为了方便使用者能够快速的跑通每一个模型,我们在每个模型下都提供了样例数据。同时,我们提供了全量数据生成的脚本,将会自动下载amazon eletronics dataset全量数据集并转换为模型能接受的
107109
输入格式。

models/rank/dien/net.py

+1
Original file line numberDiff line numberDiff line change
@@ -414,6 +414,7 @@ def __init__(self, item_emb_size, cat_emb_size, act, is_sparse,
414414
input_size=self.item_emb_size + self.cat_emb_size,
415415
hidden_size=self.item_emb_size + self.cat_emb_size,
416416
num_layers=2)
417+
# paddle暂不支持AUGRU,这里用的普通GRU,不同于论文部分
417418
self.gru_cell_attention = paddle.nn.GRUCell(
418419
self.item_emb_size + self.cat_emb_size,
419420
self.item_emb_size + self.cat_emb_size)

0 commit comments

Comments
 (0)