提交 89799b5f 编写于 作者: T tangwei

Merge branch 'develop' of ssh://gitlab.baidu.com:8022/tangwei12/paddlerec into rec_master

......@@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
evaluate:
workspace: "paddlerec.models.recall.multiview-simnet"
workspace: "paddlerec.models.match.multiview-simnet"
reader:
batch_size: 2
class: "{workspace}/evaluate_reader.py"
......@@ -24,7 +24,7 @@ train:
strategy: "async"
epochs: 2
workspace: "paddlerec.models.recall.multiview-simnet"
workspace: "paddlerec.models.match.multiview-simnet"
reader:
batch_size: 2
......
......@@ -9,13 +9,7 @@
* [整体介绍](#整体介绍)
* [匹配模型列表](#匹配模型列表)
* [使用教程](#使用教程)
* [数据处理](#数据处理)
* [训练](#训练)
* [预测](#预测)
* [效果对比](#效果对比)
* [模型效果列表](#模型效果列表)
* [分布式](#分布式)
* [模型性能列表](#模型性能列表)
* [训练&预测](#训练&预测)
## 整体介绍
### 匹配模型列表
......@@ -38,21 +32,9 @@
<p>
## 使用教程
### 数据处理
### 训练
### 预测
## 效果对比
### 模型效果列表
| 数据集 | 模型 | loss | auc |
| :------------------: | :--------------------: | :---------: |:---------: |
| - | DSSM | -- | -- |
| - | MultiView-Simnet | -- | -- |
## 分布式
### 模型性能列表
| 数据集 | 模型 | 单机 | 多机(同步) | 多机(异步) | GPU |
| :------------------: | :--------------------: | :---------: |:---------: |:---------: |:---------: |
| - | DSSM | -- | -- | -- | -- |
| - | MultiView-Simnet | -- | -- | -- | -- |
### 训练&预测
```shell
python -m paddlerec.run -m paddlerec.models.match.dssm # dssm
python -m paddlerec.run -m paddlerec.models.match.multiview-simnet # multiview-simnet
```
......@@ -9,8 +9,7 @@
* [整体介绍](#整体介绍)
* [多任务模型列表](#多任务模型列表)
* [使用教程](#使用教程)
* [训练](#训练)
* [预测](#预测)
* [训练&预测](#训练&预测)
* [效果对比](#效果对比)
* [模型效果列表](#模型效果列表)
......@@ -42,16 +41,17 @@
<p>
## 使用教程
### 训练
### 训练&预测
```shell
python -m paddlerec.run -m config.yaml -d cpu -e single
python -m paddlerec.run -m paddlerec.models.multitask.mmoe # mmoe
python -m paddlerec.run -m paddlerec.models.multitask.share-bottom # share-bottom
python -m paddlerec.run -m paddlerec.models.multitask.esmm # esmm
```
### 预测
## 效果对比
### 模型效果列表
| 数据集 | 模型 | loss | 评价指标 |
| 数据集 | 模型 | loss | auc |
| :------------------: | :--------------------: | :---------: |:---------: |
| Census-income Data | Share-Bottom | -- | 0.93120/0.99256 |
| Census-income Data | MMoE | -- | 0.94465/0.99324 |
......
......@@ -58,8 +58,15 @@
## 使用教程
### 数据处理
参考每个模型目录数据下载&预处理脚本
### 训练
```
python -m paddlerec.run -m paddlerec.models.rank.dnn # 以DNN为例
```
### 预测
```
python -m paddlerec.run -m paddlerec.models.rank.dnn # 以DNN为例
```
## 效果对比
### 模型效果 (测试)
......@@ -69,7 +76,7 @@
| Criteo | DNN | -- | 0.79395 | -- | -- |
| Criteo | DeepFM | 0.44797 | 0.80460 | -- | -- |
| Criteo | DCN | 0.44704 | 0.80654 | -- | -- |
| Criteo | xDeepFM | -- | -- | 0.48657 | -- |
| Criteo | xDeepFM | 0.48657 | -- | -- | -- |
| Census-income Data | Wide&Deep | 0.76195 | 0.90577 | -- | -- |
| Amazon Product | DIN | 0.47005 | 0.86379 | -- | -- |
......
......@@ -9,13 +9,9 @@
* [整体介绍](#整体介绍)
* [召回模型列表](#召回模型列表)
* [使用教程](#使用教程)
* [数据处理](#数据处理)
* [训练](#训练)
* [预测](#预测)
* [训练&预测](#训练&预测)
* [效果对比](#效果对比)
* [模型效果列表](#模型效果列表)
* [分布式](#分布式)
* [模型性能列表](#模型性能列表)
## 整体介绍
### 召回模型列表
......@@ -50,27 +46,20 @@
<p>
## 使用教程
### 数据处理
### 训练
### 预测
### 训练&预测
```shell
python -m paddlerec.run -m paddlerec.models.recall.word2vec # word2vec
python -m paddlerec.run -m paddlerec.models.recall.ssr # ssr
python -m paddlerec.run -m paddlerec.models.recall.gru4rec # gru4rec
python -m paddlerec.run -m paddlerec.models.recall.gnn # gnn
```
## 效果对比
### 模型效果列表
| 数据集 | 模型 | loss | Recall@20 |
| :------------------: | :--------------------: | :---------: |:---------: |
| DIGINETICA | GNN | -- | 0.507 |
| RSC15 | GRU4REC | -- | 0.67 |
| RSC15 | SSR | -- | |
| RSC15 | GRU4REC | -- | 0.670 |
| RSC15 | SSR | -- | 0.590 |
| 1 Billion Word Language Model Benchmark | Word2Vec | -- | 0.54 |
## 分布式
### 模型性能列表
| 数据集 | 模型 | 单机 | 多机(同步) | 多机(异步) | GPU |
| :------------------: | :--------------------: | :---------: |:---------: |:---------: |:---------: |
| DIGINETICA | GNN | -- | -- | -- | -- |
| RSC15 | GRU4REC | -- | -- | -- | -- |
| RSC15 | SSR | -- | -- | -- | -- |
| 1 Billion Word Language Model Benchmark | Word2Vec | -- | -- | -- | -- |
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册