readme.md 6.3 KB
Newer Older
C
chengmo 已提交
1
<p align="center">
C
chengmo 已提交
2
<img align="center" src="doc/imgs/logo.png">
C
chengmo 已提交
3 4
<p>

C
chengmo 已提交
5 6 7 8 9 10 11 12 13 14 15
<p align="center">
    <br>
    <a href="https://travis-ci.com/PaddlePaddle/Serving">
        <img alt="Build Status" src="https://img.shields.io/travis/com/PaddlePaddle/Serving/develop">
    </a>
    <img alt="Release" src="https://img.shields.io/badge/Release-0.0.3-yellowgreen">
    <img alt="Issues" src="https://img.shields.io/github/issues/PaddlePaddle/Serving">
    <img alt="License" src="https://img.shields.io/github/license/PaddlePaddle/Serving">
    <img alt="Slack" src="https://img.shields.io/badge/Join-Slack-green">
    <br>
<p>
C
chengmo 已提交
16 17


C
chengmo 已提交
18
<h2 align="center">什么是PaddleRec</h2>
C
chengmo 已提交
19 20

<p align="center">
C
chengmo 已提交
21
<img align="center" src="doc/imgs/structure.png">
C
chengmo 已提交
22 23
<p>

C
chengmo 已提交
24 25 26 27
- 源于飞桨生态的`搜索推荐模型`**一站式开箱即用工具** 
- 适合初学者,开发者,研究者的调研,训练到预测部署的全流程解决方案
- 包含语义理解、召回、粗排、精排、多任务学习、融合等多个任务的推荐搜索算法库
- 自定义`yaml`即可快速上手使用单机训练、大规模分布式训练、离线预测、在线部署
C
chengmo 已提交
28

C
chengmo 已提交
29

C
chengmo 已提交
30 31 32 33 34 35
<h2 align="center">PadlleRec概览</h2>

<p align="center">
<img align="center" src="doc/imgs/overview.png">
<p>

C
chengmo 已提交
36

C
chengmo 已提交
37
<h2 align="center">便捷安装</h2>
C
chengmo 已提交
38

C
chengmo 已提交
39
### 环境要求
C
chengmo 已提交
40 41
* Python 2.7/ 3.5 / 3.6 / 3.7
* PaddlePaddle  >= 1.7.2
C
update  
chengmo 已提交
42
* 操作系统: Windows/Mac/Linux
C
fix  
chengmo 已提交
43
  
C
chengmo 已提交
44
### 安装命令
C
fix  
chengmo 已提交
45

C
update  
chengmo 已提交
46
- 安装方法一<PIP源直接安装>
C
fix  
chengmo 已提交
47
  ```bash
48
  python -m pip install paddle-rec
C
fix  
chengmo 已提交
49
  ```
C
update  
chengmo 已提交
50 51 52

- 安装方法二

C
chengmo 已提交
53
  源码编译安装
C
chengmo 已提交
54
  1. 安装飞桨  **注:需要用户安装版本 >1.7.2 的飞桨**
C
update  
chengmo 已提交
55

C
chengmo 已提交
56
    ```shell
C
fix  
chengmo 已提交
57 58
    python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple
    ```
C
update  
chengmo 已提交
59

C
chengmo 已提交
60
  2. 源码安装Fleet-Rec
C
update  
chengmo 已提交
61

C
fix  
chengmo 已提交
62
    ```
63 64
    git clone https://github.com/PaddlePaddle/PaddleRec/
    cd PaddleRec
C
fix  
chengmo 已提交
65 66
    python setup.py install
    ```
C
update  
chengmo 已提交
67

C
chengmo 已提交
68

C
chengmo 已提交
69 70 71 72
<h2 align="center">快速启动</h2>



C
chengmo 已提交
73 74 75 76

### 启动内置模型

目前框架内置了多个模型,简单的命令即可使用内置模型开始单机训练和本地1*1模拟训练,我们以`ctr-dnn`为例介绍PaddleRec的简单使用。
C
chengmo 已提交
77

C
chengmo 已提交
78
<h3 align="center">单机训练</h3>
C
update  
chengmo 已提交
79 80

```bash
C
chengmo 已提交
81 82 83 84 85
# 使用CPU进行单机训练
python -m fleetrec.run -m fleetrec.models.rank.dnn -d cpu -e single 

# 使用GPU进行单机训练
python -m fleetrec.run -m fleetrec.models.rank.dnn -d gpu -e single
C
update  
chengmo 已提交
86 87
```

C
chengmo 已提交
88
<h3 align="center">本地模拟分布式训练</h3>
C
update  
chengmo 已提交
89 90

```bash
C
chengmo 已提交
91 92
# 使用CPU资源进行本地模拟分布式训练
python -m fleetrec.run -m fleetrec.models.rank.dnn -d cpu -e local_cluster
C
update  
chengmo 已提交
93 94
```

C
chengmo 已提交
95
<h3 align="center">集群分布式训练</h3>
C
update  
chengmo 已提交
96 97

```bash
C
chengmo 已提交
98 99
# 配置好 mpi/k8s/paddlecloud集群环境后
python -m fleetrec.run -m fleetrec.models.rank.dnn -d cpu -e cluster
C
update  
chengmo 已提交
100 101
```

C
chengmo 已提交
102 103 104 105
### 启动自定义模型

若您复用内置模型,更改了超参,重新配置了数据后

C
chengmo 已提交
106 107
<h2 align="center">支持模型列表</h2>

C
chengmo 已提交
108 109
> 部分表格占位待改(大规模稀疏)

C
chengmo 已提交
110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127
|   方向   |                                      模型                                      | 单机CPU训练 | 单机GPU训练 | 分布式CPU训练 | 分布式GPU训练 |
| :------: | :----------------------------------------------------------------------------: | :---------: | :---------: | :-----------: | :-----------: |
| 内容理解 | [Text-Classifcation](models/contentunderstanding/text_classification/model.py) |      ✓      |      x      |       ✓       |       x       |
| 内容理解 |           [TagSpace](models/contentunderstanding/tagspace/model.py)            |      ✓      |      x      |       ✓       |       x       |
|   召回   |                  [Word2Vec](models/recall/word2vec/model.py)                   |      ✓      |      x      |       ✓       |       x       |
|   召回   |                       [TDM](models/recall/tdm/model.py)                        |      ✓      |      x      |       ✓       |       x       |
|   召回   |                       [SSR](models/recall/ssr/model.py)                        |      ✓      |      ✓      |       ✓       |       x       |
|   召回   |                   [Gru4Rec](models/recall/gru4rec/model.py)                    |      ✓      |      ✓      |       ✓       |       x       |
|   排序   |                      [CTR-Dnn](models/rank/dnn/model.py)                       |      ✓      |      x      |       ✓       |       x       |
|   排序   |                     [DeepFm](models/rank/deepfm/model.py)                      |      ✓      |      x      |       ✓       |       x       |
|   排序   |                    [xDeepFm](models/rank/xdeepfm/model.py)                     |      ✓      |      x      |       ✓       |       x       |
|   排序   |                        [DIN](models/rank/din/model.py)                         |      ✓      |      x      |       ✓       |       x       |
|   排序   |                  [Wide&Deep](models/rank/wide_deep/model.py)                   |      ✓      |      x      |       ✓       |       x       |
|  多任务  |                     [ESMM](models/multitask/essm/model.py)                     |      ✓      |      ✓      |       ✓       |       x       |
|  多任务  |                     [MMOE](models/multitask/mmoe/model.py)                     |      ✓      |      ✓      |       ✓       |       x       |
|   排序   |             [ShareBottom](models/multitask/share-bottom/model.py)              |      ✓      |      ✓      |       ✓       |       x       |
|   匹配   |                       [DSSM](models/match/dssm/model.py)                       |      ✓      |      x      |       ✓       |       x       |
|   匹配   |                [Simnet](models/match/multiview-simnet/model.py)                |      ✓      |      x      |       ✓       |       x       |
C
chengmo 已提交
128 129 130



C
chengmo 已提交
131 132 133 134 135
<h2 align="center">文档</h2>

### 新手教程
* [环境要求](#环境要求)
* [安装命令](#安装命令)
C
chengmo 已提交
136
* [快速开始](#一行命令启动训练)
C
chengmo 已提交
137 138
* [推荐系统背景知识](doc/)

C
chengmo 已提交
139
### 进阶教程
C
chengmo 已提交
140 141
* [自定义数据集及Reader](doc/custom_dataset_reader.md)
* [分布式训练](doc/distributed_train.md)
C
chengmo 已提交
142 143

### 关于PaddleRec性能
C
chengmo 已提交
144
* [Benchmark](doc/benchmark.md)
C
chengmo 已提交
145

C
chengmo 已提交
146
### FAQ
C
chengmo 已提交
147
* [常见问题FAQ](doc/faq.md)
T
tangwei 已提交
148

C
chengmo 已提交
149

C
chengmo 已提交
150
<h2 align="center">社区</h2>
C
chengmo 已提交
151

C
chengmo 已提交
152 153 154 155
### 反馈
如有意见、建议及使用中的BUG,欢迎在`GitHub Issue`提交

### 版本历史
C
chengmo 已提交
156
- 2020.5.14 - PaddleRec v0.1
C
chengmo 已提交
157 158 159 160
  
### 许可证书
本项目的发布受[Apache 2.0 license](LICENSE)许可认证。