readme.md 3.1 KB
Newer Older
M
malin10 已提交
1
# 匹配模型库
M
malin10 已提交
2 3

## 简介
Y
yinhaofeng 已提交
4
我们提供了常见的匹配任务中使用的模型算法的PaddleRec实现, 单机训练&预测效果指标以及分布式训练&预测性能指标等。实现的模型包括 [DSSM](http://gitlab.baidu.com/tangwei12/paddlerec/tree/develop/models/match/dssm)[MultiView-Simnet](http://gitlab.baidu.com/tangwei12/paddlerec/tree/develop/models/match/multiview-simnet)[match-pyramid](https://github.com/PaddlePaddle/PaddleRec/tree/master/models/match/match-pyramid)
M
malin10 已提交
5 6 7 8 9 10 11

模型算法库在持续添加中,欢迎关注。

## 目录
* [整体介绍](#整体介绍)
    * [匹配模型列表](#匹配模型列表)
* [使用教程](#使用教程)
M
malin10 已提交
12
    * [训练&预测](#训练&预测)
M
malin10 已提交
13 14 15 16 17 18

## 整体介绍
### 匹配模型列表

|       模型        |       简介        |       论文        |
| :------------------: | :--------------------: | :---------: |
C
Chengmo 已提交
19 20
| DSSM | Deep Structured Semantic Models | [CIKM 2013][Learning Deep Structured Semantic Models for Web Search using Clickthrough Data](https://www.microsoft.com/en-us/research/wp-content/uploads/2016/02/cikm2013_DSSM_fullversion.pdf) |
| MultiView-Simnet | Multi-view Simnet for Personalized recommendation | [WWW 2015][A Multi-View Deep Learning Approach for Cross Domain User Modeling in Recommendation Systems](https://www.microsoft.com/en-us/research/wp-content/uploads/2016/02/frp1159-songA.pdf) |
Y
yinhaofeng 已提交
21 22
| match-pyramid | Text Matching as Image Recognition | [arXiv W2016][Text Matching as Image Recognition](https://arxiv.org/pdf/1602.06359.pdf) |

M
malin10 已提交
23

M
malin10 已提交
24 25 26 27 28 29 30 31 32 33 34 35
下面是每个模型的简介(注:图片引用自链接中的论文)

[DSSM](https://www.microsoft.com/en-us/research/wp-content/uploads/2016/02/cikm2013_DSSM_fullversion.pdf):
<p align="center">
<img align="center" src="../../doc/imgs/dssm.png">
<p>

[MultiView-Simnet](https://www.microsoft.com/en-us/research/wp-content/uploads/2016/02/frp1159-songA.pdf):
<p align="center">
<img align="center" src="../../doc/imgs/multiview-simnet.png">
<p>

Y
yinhaofeng 已提交
36 37 38 39
[match-pyramid](https://arxiv.org/pdf/1602.06359.pdf):
<p align="center">
<img align="center" src="../../doc/imgs/match-pyramid.png">
<p>
C
Chengmo 已提交
40

Y
yinhaofeng 已提交
41 42 43 44
## 使用教程(快速开始)
### 训练&预测
每个模型都提供了样例数据可以供您快速体验,在paddlerec目录下直接执行下面的命令即可启动训练:
```
C
Chengmo 已提交
45 46
python -m paddlerec.run -m models/match/dssm/config.yaml # dssm
python -m paddlerec.run -m models/match/multiview-simnet/config.yaml # multiview-simnet
Y
yinhaofeng 已提交
47
python -m paddlerec.run -m models/contentunderstanding/match-pyramid/config.yaml #match-pyramid
M
malin10 已提交
48
```
Y
yinhaofeng 已提交
49 50
### 效果复现
每个模型下的readme中都有详细的效果复现的教程,您可以进入模型的目录中详细查看  
M
malin10 已提交
51

Y
yinhaofeng 已提交
52
### 模型效果 (测试)
M
malin10 已提交
53

Y
yinhaofeng 已提交
54 55 56 57 58
|       数据集        |       模型       |      auc          |       map       |  
| :------------------: | :--------------------: | :---------: |:---------: |
|       zhidao       |       DSSM       |       0.55        |       --          | 
|       Letor07        |       match-pyramid       |       --        |      0.42          | 
|       zhidao        |       multiview-simnet       |       0.59        |       --          |