README.cn.md 29.5 KB
Newer Older
C
choijulie 已提交
1 2
# 个性化推荐

T
tink2123 已提交
3
本教程源代码目录在[book/recommender_system](https://github.com/PaddlePaddle/book/tree/develop/05.recommender_system),初次使用请您参考[Book文档使用说明](https://github.com/PaddlePaddle/book/blob/develop/README.cn.md#运行这本书)
C
choijulie 已提交
4

5 6 7 8 9 10 11 12
### 说明: ###
1. 硬件环境要求:
本文可支持在CPU、GPU下运行
2. Docker镜像支持的CUDA/cuDNN版本:
如果使用了Docker运行Book,请注意:这里所提供的默认镜像的GPU环境为 CUDA 8/cuDNN 5,对于NVIDIA Tesla V100等要求CUDA 9的 GPU,使用该镜像可能会运行失败。
3. 文档和脚本中代码的一致性问题:
请注意:为使本文更加易读易用,我们拆分、调整了train.py的代码并放入本文。本文中代码与train.py的运行结果一致,可直接运行[train.py](https://github.com/PaddlePaddle/book/blob/develop/05.recommender_system/train.py)进行验证。

C
choijulie 已提交
13 14
## 背景介绍

15
在网络技术不断发展和电子商务规模不断扩大的背景下,商品数量和种类快速增长,用户需要花费大量时间才能找到自己想买的商品,这就是信息超载问题。为了解决这个难题,个性化推荐系统(Recommender System)应运而生。
C
choijulie 已提交
16

17
个性化推荐系统是信息过滤系统(Information Filtering System)的子集,它可以用在很多领域,如电影、音乐、电商和 Feed 流推荐等。个性化推荐系统通过分析、挖掘用户行为,发现用户的个性化需求与兴趣特点,将用户可能感兴趣的信息或商品推荐给用户。与搜索引擎不同,个性化推荐系统不需要用户准确地描述出自己的需求,而是根据用户的历史行为进行建模,主动提供满足用户兴趣和需求的信息。
C
choijulie 已提交
18

19
1994年明尼苏达大学推出的GroupLens系统[[1](#参考文献)]一般被认为是个性化推荐系统成为一个相对独立的研究方向的标志。该系统首次提出了基于协同过滤来完成推荐任务的思想,此后,基于该模型的协同过滤推荐引领了个性化推荐系统十几年的发展方向。
C
choijulie 已提交
20

21
传统的个性化推荐系统方法主要有:
C
choijulie 已提交
22

23 24 25
- 协同过滤推荐(Collaborative Filtering Recommendation):该方法是应用最广泛的技术之一,需要收集和分析用户的历史行为、活动和偏好。它通常可以分为两个子类:基于用户 (User-Based)的推荐[[1](#参考文献)] 和基于物品(Item-Based)的推荐[[2](#参考文献)]。该方法的一个关键优势是它不依赖于机器去分析物品的内容特征,因此它无需理解物品本身也能够准确地推荐诸如电影之类的复杂物品;缺点是对于没有任何行为的新用户存在冷启动的问题,同时也存在用户与商品之间的交互数据不够多造成的稀疏问题。值得一提的是,社交网络[[3](#参考文献)]或地理位置等上下文信息都可以结合到协同过滤中去。
- 基于内容过滤推荐[[4](#参考文献)](Content-based Filtering Recommendation):该方法利用商品的内容描述,抽象出有意义的特征,通过计算用户的兴趣和商品描述之间的相似度,来给用户做推荐。优点是简单直接,不需要依据其他用户对商品的评价,而是通过商品属性进行商品相似度度量,从而推荐给用户所感兴趣商品的相似商品;缺点是对于没有任何行为的新用户同样存在冷启动的问题。
- 组合推荐[[5](#参考文献)](Hybrid Recommendation):运用不同的输入和技术共同进行推荐,以弥补各自推荐技术的缺点。
C
choijulie 已提交
26

27
近些年来,深度学习在很多领域都取得了巨大的成功。学术界和工业界都在尝试将深度学习应用于个性化推荐系统领域中。深度学习具有优秀的自动提取特征的能力,能够学习多层次的抽象特征表示,并对异质或跨域的内容信息进行学习,可以一定程度上处理个性化推荐系统冷启动问题[[6](#参考文献)]。本教程主要介绍个性化推荐的深度学习模型,以及如何使用PaddlePaddle实现模型。
C
choijulie 已提交
28 29 30 31 32 33 34 35 36 37 38 39 40

## 效果展示

我们使用包含用户信息、电影信息与电影评分的数据集作为个性化推荐的应用场景。当我们训练好模型后,只需要输入对应的用户ID和电影ID,就可以得出一个匹配的分数(范围[0,5],分数越高视为兴趣越大),然后根据所有电影的推荐得分排序,推荐给用户可能感兴趣的电影。

```
Input movie_id: 1962
Input user_id: 1
Prediction Score is 4.25
```

## 模型概览

41
本章中,我们首先介绍YouTube的视频个性化推荐系统[[7](#参考文献)],然后介绍我们实现的融合推荐模型。
C
choijulie 已提交
42

43
### YouTube的深度神经网络个性化推荐系统
C
choijulie 已提交
44

45
YouTube是世界上最大的视频上传、分享和发现网站,YouTube个性化推荐系统为超过10亿用户从不断增长的视频库中推荐个性化的内容。整个系统由两个神经网络组成:候选生成网络和排序网络。候选生成网络从百万量级的视频库中生成上百个候选,排序网络对候选进行打分排序,输出排名最高的数十个结果。系统结构如图1所示:
C
choijulie 已提交
46 47

<p align="center">
48
<img src="https://github.com/PaddlePaddle/book/blob/develop/05.recommender_system/image/YouTube_Overview.png?raw=true" width="70%" ><br/>
49
图1. YouTube 个性化推荐系统结构
C
choijulie 已提交
50 51 52 53 54 55
</p>

#### 候选生成网络(Candidate Generation Network)

候选生成网络将推荐问题建模为一个类别数极大的多类分类问题:对于一个Youtube用户,使用其观看历史(视频ID)、搜索词记录(search tokens)、人口学信息(如地理位置、用户登录设备)、二值特征(如性别,是否登录)和连续特征(如用户年龄)等,对视频库中所有视频进行多分类,得到每一类别的分类结果(即每一个视频的推荐概率),最终输出概率较高的几百个视频。

56
首先,将观看历史及搜索词记录这类历史信息,映射为向量后取平均值得到定长表示;同时,输入人口学特征以优化新用户的推荐效果,并将二值特征和连续特征归一化处理到[0, 1]范围。接下来,将所有特征表示拼接为一个向量,并输入给非线形多层感知器(MLP,详见[识别数字](https://github.com/PaddlePaddle/book/blob/develop/02.recognize_digits/README.cn.md)教程)处理。最后,训练时将MLP的输出给softmax做分类,预测时计算用户的综合特征(MLP的输出)与所有视频的相似度,取得分最高的$k$个作为候选生成网络的筛选结果。图2显示了候选生成网络结构。
C
choijulie 已提交
57 58

<p align="center">
59
<img src="https://github.com/PaddlePaddle/book/blob/develop/05.recommender_system/image/Deep_candidate_generation_model_architecture.png?raw=true" width="70%" ><br/>
C
choijulie 已提交
60 61 62 63 64
图2. 候选生成网络结构
</p>

对于一个用户$U$,预测此刻用户要观看的视频$\omega$为视频$i$的概率公式为:

65 66 67
<p align="center">
<img src="https://github.com/PaddlePaddle/book/blob/develop/05.recommender_system/image/formula1.png?raw=true" width="20%" ><br/>
</p>
C
choijulie 已提交
68 69 70 71 72 73 74 75 76

其中$u$为用户$U$的特征表示,$V$为视频库集合,$v_i$为视频库中第$i$个视频的特征表示。$u$和$v_i$为长度相等的向量,两者点积可以通过全连接层实现。

考虑到softmax分类的类别数非常多,为了保证一定的计算效率:1)训练阶段,使用负样本类别采样将实际计算的类别数缩小至数千;2)推荐(预测)阶段,忽略softmax的归一化计算(不影响结果),将类别打分问题简化为点积(dot product)空间中的最近邻(nearest neighbor)搜索问题,取与$u$最近的$k$个视频作为生成的候选。

#### 排序网络(Ranking Network)
排序网络的结构类似于候选生成网络,但是它的目标是对候选进行更细致的打分排序。和传统广告排序中的特征抽取方法类似,这里也构造了大量的用于视频排序的相关特征(如视频 ID、上次观看时间等)。这些特征的处理方式和候选生成网络类似,不同之处是排序网络的顶部是一个加权逻辑回归(weighted logistic regression),它对所有候选视频进行打分,从高到底排序后将分数较高的一些视频返回给用户。

### 融合推荐模型
77
本节会使用卷积神经网络(Convolutional Neural Networks)来学习电影名称的表示。下面会依次介绍文本卷积神经网络以及融合推荐模型。
C
choijulie 已提交
78 79 80

#### 文本卷积神经网络(CNN)

81
卷积神经网络经常用来处理具有类似网格拓扑结构(grid-like topology)的数据。例如,图像可以视为二维网格的像素点,自然语言可以视为一维的词序列。卷积神经网络可以提取多种局部特征,并对其进行组合抽象得到更高级的特征表示。实验表明,卷积神经网络能高效地对图像及文本问题进行建模处理。
C
choijulie 已提交
82 83 84 85

卷积神经网络主要由卷积(convolution)和池化(pooling)操作构成,其应用及组合方式灵活多变,种类繁多。本小结我们以如图3所示的网络进行讲解:

<p align="center">
86
<img src="https://github.com/PaddlePaddle/book/blob/develop/05.recommender_system/image/text_cnn.png?raw=true" width = "80%" align="center"/><br/>
C
choijulie 已提交
87 88 89
图3. 卷积神经网络文本分类模型
</p>

90
假设待处理句子的长度为$n$,其中第$i$个词的词向量为$x_i\in\mathbb{R}^k$,$k$为维度大小。
C
choijulie 已提交
91

92
首先,进行词向量的拼接操作:将每$h$个词拼接起来形成一个大小为$h$的词窗口,记为$x_{i:i+h-1}$,它表示词序列$x_{i},x_{i+1},\ldots,x_{i+h-1}$的拼接,其中,$i$表示词窗口中第一个词在整个句子中的位置,取值范围从$1$到$n-h+1$,$x_{i:i+h-1}\in\mathbb{R}^{hk}$。
C
choijulie 已提交
93 94 95

其次,进行卷积操作:把卷积核(kernel)$w\in\mathbb{R}^{hk}$应用于包含$h$个词的窗口$x_{i:i+h-1}$,得到特征$c_i=f(w\cdot x_{i:i+h-1}+b)$,其中$b\in\mathbb{R}$为偏置项(bias),$f$为非线性激活函数,如$sigmoid$。将卷积核应用于句子中所有的词窗口${x_{1:h},x_{2:h+1},\ldots,x_{n-h+1:n}}$,产生一个特征图(feature map):

96 97 98
<p align="center">
<img src="https://github.com/PaddlePaddle/book/blob/develop/05.recommender_system/image/formula2.png?raw=true" width="40%" ><br/>
</p>
C
choijulie 已提交
99 100 101

接下来,对特征图采用时间维度上的最大池化(max pooling over time)操作得到此卷积核对应的整句话的特征$\hat c$,它是特征图中所有元素的最大值:

102 103 104
<p align="center">
<img src="https://github.com/PaddlePaddle/book/blob/develop/05.recommender_system/image/formula3.png?raw=true" width="15%" ><br/>
</p>
C
choijulie 已提交
105

106
#### 融合推荐模型概览
C
choijulie 已提交
107

108
在融合推荐模型的电影个性化推荐系统中:
C
choijulie 已提交
109 110 111 112 113 114 115 116 117 118 119

1. 首先,使用用户特征和电影特征作为神经网络的输入,其中:

   - 用户特征融合了四个属性信息,分别是用户ID、性别、职业和年龄。

   - 电影特征融合了三个属性信息,分别是电影ID、电影类型ID和电影名称。

2. 对用户特征,将用户ID映射为维度大小为256的向量表示,输入全连接层,并对其他三个属性也做类似的处理。然后将四个属性的特征表示分别全连接并相加。

3. 对电影特征,将电影ID以类似用户ID的方式进行处理,电影类型ID以向量的形式直接输入全连接层,电影名称用文本卷积神经网络得到其定长向量表示。然后将三个属性的特征表示分别全连接并相加。

120
4. 得到用户和电影的向量表示后,计算二者的余弦相似度作为个性化推荐系统的打分。最后,用该相似度打分和用户真实打分的差异的平方作为该回归模型的损失函数。
C
choijulie 已提交
121 122

<p align="center">
123
<img src="https://github.com/PaddlePaddle/book/blob/develop/05.recommender_system/image/rec_regression_network.png?raw=true" width="90%" ><br/>
C
choijulie 已提交
124 125 126 127 128 129 130 131 132 133 134 135 136
图4. 融合推荐模型
</p>

## 数据准备

### 数据介绍与下载

我们以 [MovieLens 百万数据集(ml-1m)](http://files.grouplens.org/datasets/movielens/ml-1m.zip)为例进行介绍。ml-1m 数据集包含了 6,000 位用户对 4,000 部电影的 1,000,000 条评价(评分范围 1~5 分,均为整数),由 GroupLens Research 实验室搜集整理。

Paddle在API中提供了自动加载数据的模块。数据模块为 `paddle.dataset.movielens`


```python
137
from __future__ import print_function
D
daminglu 已提交
138 139
import paddle
movie_info = paddle.dataset.movielens.movie_info()
140
print(list(movie_info.values())[0])
C
choijulie 已提交
141 142 143 144 145 146 147 148 149 150 151 152 153 154 155
```


```python
# Run this block to show dataset's documentation
# help(paddle.dataset.movielens)
```

在原始数据中包含电影的特征数据,用户的特征数据,和用户对电影的评分。

例如,其中某一个电影特征为:


```python
movie_info = paddle.dataset.movielens.movie_info()
156
print(list(movie_info.values())[0])
C
choijulie 已提交
157 158 159 160 161 162 163 164 165 166
```

    <MovieInfo id(1), title(Toy Story ), categories(['Animation', "Children's", 'Comedy'])>


这表示,电影的id是1,标题是《Toy Story》,该电影被分为到三个类别中。这三个类别是动画,儿童,喜剧。


```python
user_info = paddle.dataset.movielens.user_info()
167
print(list(user_info.values())[0])
C
choijulie 已提交
168 169 170 171 172 173 174 175 176
```

    <UserInfo id(1), gender(F), age(1), job(10)>


这表示,该用户ID是1,女性,年龄比18岁还年轻。职业ID是10。


其中,年龄使用下列分布
177

C
choijulie 已提交
178 179 180 181 182 183 184 185 186
*  1:  "Under 18"
* 18:  "18-24"
* 25:  "25-34"
* 35:  "35-44"
* 45:  "45-49"
* 50:  "50-55"
* 56:  "56+"

职业是从下面几种选项里面选则得出:
187

C
choijulie 已提交
188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219
*  0:  "other" or not specified
*  1:  "academic/educator"
*  2:  "artist"
*  3:  "clerical/admin"
*  4:  "college/grad student"
*  5:  "customer service"
*  6:  "doctor/health care"
*  7:  "executive/managerial"
*  8:  "farmer"
*  9:  "homemaker"
* 10:  "K-12 student"
* 11:  "lawyer"
* 12:  "programmer"
* 13:  "retired"
* 14:  "sales/marketing"
* 15:  "scientist"
* 16:  "self-employed"
* 17:  "technician/engineer"
* 18:  "tradesman/craftsman"
* 19:  "unemployed"
* 20:  "writer"

而对于每一条训练/测试数据,均为 <用户特征> + <电影特征> + 评分。

例如,我们获得第一条训练数据:


```python
train_set_creator = paddle.dataset.movielens.train()
train_sample = next(train_set_creator())
uid = train_sample[0]
mov_id = train_sample[len(user_info[uid].value())]
220
print ("User %s rates Movie %s with Score %s"%(user_info[uid], movie_info[mov_id], train_sample[-1]))
C
choijulie 已提交
221 222 223 224 225 226 227 228 229
```

    User <UserInfo id(1), gender(F), age(1), job(10)> rates Movie <MovieInfo id(1193), title(One Flew Over the Cuckoo's Nest ), categories(['Drama'])> with Score [5.0]


即用户1对电影1193的评价为5分。

## 模型配置说明

D
daminglu 已提交
230
下面我们开始根据输入数据的形式配置模型。首先引入所需的库函数以及定义全局变量。
231 232
- IS_SPARSE: embedding中是否使用稀疏更新
- PASS_NUM: epoch数量
C
choijulie 已提交
233 234 235


```python
D
daminglu 已提交
236 237 238 239 240 241 242 243 244 245
import math
import sys
import numpy as np
import paddle
import paddle.fluid as fluid
import paddle.fluid.layers as layers
import paddle.fluid.nets as nets

IS_SPARSE = True
BATCH_SIZE = 256
246
PASS_NUM = 20
C
choijulie 已提交
247 248
```

D
daminglu 已提交
249
然后为我们的用户特征综合模型定义模型配置
C
choijulie 已提交
250 251

```python
D
daminglu 已提交
252
def get_usr_combined_features():
253
    """network definition for user part"""
D
daminglu 已提交
254 255 256

    USR_DICT_SIZE = paddle.dataset.movielens.max_user_id() + 1

H
hutuxian 已提交
257
    uid = fluid.data(name='user_id', shape=[-1], dtype='int64')
D
daminglu 已提交
258

H
hutuxian 已提交
259
    usr_emb = fluid.embedding(
D
daminglu 已提交
260 261 262 263 264 265 266 267 268 269
        input=uid,
        dtype='float32',
        size=[USR_DICT_SIZE, 32],
        param_attr='user_table',
        is_sparse=IS_SPARSE)

    usr_fc = layers.fc(input=usr_emb, size=32)

    USR_GENDER_DICT_SIZE = 2

H
hutuxian 已提交
270
    usr_gender_id = fluid.data(name='gender_id', shape=[-1], dtype='int64')
D
daminglu 已提交
271

H
hutuxian 已提交
272
    usr_gender_emb = fluid.embedding(
D
daminglu 已提交
273 274 275 276 277 278 279 280
        input=usr_gender_id,
        size=[USR_GENDER_DICT_SIZE, 16],
        param_attr='gender_table',
        is_sparse=IS_SPARSE)

    usr_gender_fc = layers.fc(input=usr_gender_emb, size=16)

    USR_AGE_DICT_SIZE = len(paddle.dataset.movielens.age_table)
H
hutuxian 已提交
281
    usr_age_id = fluid.data(name='age_id', shape=[-1], dtype="int64")
D
daminglu 已提交
282

H
hutuxian 已提交
283
    usr_age_emb = fluid.embedding(
D
daminglu 已提交
284 285 286 287 288 289 290 291
        input=usr_age_id,
        size=[USR_AGE_DICT_SIZE, 16],
        is_sparse=IS_SPARSE,
        param_attr='age_table')

    usr_age_fc = layers.fc(input=usr_age_emb, size=16)

    USR_JOB_DICT_SIZE = paddle.dataset.movielens.max_job_id() + 1
H
hutuxian 已提交
292
    usr_job_id = fluid.data(name='job_id', shape=[-1], dtype="int64")
D
daminglu 已提交
293

H
hutuxian 已提交
294
    usr_job_emb = fluid.embedding(
D
daminglu 已提交
295 296 297 298 299 300 301 302 303 304 305 306 307
        input=usr_job_id,
        size=[USR_JOB_DICT_SIZE, 16],
        param_attr='job_table',
        is_sparse=IS_SPARSE)

    usr_job_fc = layers.fc(input=usr_job_emb, size=16)

    concat_embed = layers.concat(
        input=[usr_fc, usr_gender_fc, usr_age_fc, usr_job_fc], axis=1)

    usr_combined_features = layers.fc(input=concat_embed, size=200, act="tanh")

    return usr_combined_features
C
choijulie 已提交
308 309
```

D
daminglu 已提交
310 311
如上述代码所示,对于每个用户,我们输入4维特征。其中包括user_id,gender_id,age_id,job_id。这几维特征均是简单的整数值。为了后续神经网络处理这些特征方便,我们借鉴NLP中的语言模型,将这几维离散的整数值,变换成embedding取出。分别形成usr_emb, usr_gender_emb, usr_age_emb, usr_job_emb。

C
choijulie 已提交
312 313 314 315 316 317
然后,我们对于所有的用户特征,均输入到一个全连接层(fc)中。将所有特征融合为一个200维度的特征。

进而,我们对每一个电影特征做类似的变换,网络配置为:


```python
D
daminglu 已提交
318
def get_mov_combined_features():
319
    """network definition for item(movie) part"""
C
choijulie 已提交
320

D
daminglu 已提交
321
    MOV_DICT_SIZE = paddle.dataset.movielens.max_movie_id() + 1
C
choijulie 已提交
322

H
hutuxian 已提交
323
    mov_id = fluid.data(name='movie_id', shape=[-1], dtype='int64')
C
choijulie 已提交
324

H
hutuxian 已提交
325
    mov_emb = fluid.embedding(
D
daminglu 已提交
326 327 328 329 330
        input=mov_id,
        dtype='float32',
        size=[MOV_DICT_SIZE, 32],
        param_attr='movie_table',
        is_sparse=IS_SPARSE)
C
choijulie 已提交
331

D
daminglu 已提交
332
    mov_fc = layers.fc(input=mov_emb, size=32)
C
choijulie 已提交
333

D
daminglu 已提交
334
    CATEGORY_DICT_SIZE = len(paddle.dataset.movielens.movie_categories())
C
choijulie 已提交
335

H
hutuxian 已提交
336 337
    category_id = fluid.data(
        name='category_id', shape=[-1], dtype='int64', lod_level=1)
C
choijulie 已提交
338

H
hutuxian 已提交
339
    mov_categories_emb = fluid.embedding(
D
daminglu 已提交
340
        input=category_id, size=[CATEGORY_DICT_SIZE, 32], is_sparse=IS_SPARSE)
C
choijulie 已提交
341

D
daminglu 已提交
342 343
    mov_categories_hidden = layers.sequence_pool(
        input=mov_categories_emb, pool_type="sum")
C
choijulie 已提交
344

D
daminglu 已提交
345
    MOV_TITLE_DICT_SIZE = len(paddle.dataset.movielens.get_movie_title_dict())
C
choijulie 已提交
346

H
hutuxian 已提交
347 348
    mov_title_id = fluid.data(
        name='movie_title', shape=[-1], dtype='int64', lod_level=1)
C
choijulie 已提交
349

H
hutuxian 已提交
350
    mov_title_emb = fluid.embedding(
D
daminglu 已提交
351
        input=mov_title_id, size=[MOV_TITLE_DICT_SIZE, 32], is_sparse=IS_SPARSE)
C
choijulie 已提交
352

D
daminglu 已提交
353 354 355 356 357 358 359 360 361
    mov_title_conv = nets.sequence_conv_pool(
        input=mov_title_emb,
        num_filters=32,
        filter_size=3,
        act="tanh",
        pool_type="sum")

    concat_embed = layers.concat(
        input=[mov_fc, mov_categories_hidden, mov_title_conv], axis=1)
C
choijulie 已提交
362

D
daminglu 已提交
363
    mov_combined_features = layers.fc(input=concat_embed, size=200, act="tanh")
C
choijulie 已提交
364

D
daminglu 已提交
365 366
    return mov_combined_features
```
C
choijulie 已提交
367

D
daminglu 已提交
368
电影标题名称(title)是一个序列的整数,整数代表的是这个词在索引序列中的下标。这个序列会被送入 `sequence_conv_pool` 层,这个层会在时间维度上使用卷积和池化。因为如此,所以输出会是固定长度,尽管输入的序列长度各不相同。
C
choijulie 已提交
369

D
daminglu 已提交
370
最后,我们定义一个`inference_program`来使用余弦相似度计算用户特征与电影特征的相似性。
C
choijulie 已提交
371 372

```python
D
daminglu 已提交
373
def inference_program():
374 375
    """the combined network"""

D
daminglu 已提交
376 377 378 379 380 381 382
    usr_combined_features = get_usr_combined_features()
    mov_combined_features = get_mov_combined_features()

    inference = layers.cos_sim(X=usr_combined_features, Y=mov_combined_features)
    scale_infer = layers.scale(x=inference, scale=5.0)

    return scale_infer
C
choijulie 已提交
383 384
```

D
daminglu 已提交
385 386 387 388
进而,我们定义一个`train_program`来使用`inference_program`计算出的结果,在标记数据的帮助下来计算误差。我们还定义了一个`optimizer_func`来定义优化器。

```python
def train_program():
389
    """define the cost function"""
C
choijulie 已提交
390

D
daminglu 已提交
391
    scale_infer = inference_program()
C
choijulie 已提交
392

H
hutuxian 已提交
393
    label = fluid.data(name='score', shape=[-1, 1], dtype='float32')
D
daminglu 已提交
394 395
    square_cost = layers.square_error_cost(input=scale_infer, label=label)
    avg_cost = layers.mean(square_cost)
C
choijulie 已提交
396

D
daminglu 已提交
397
    return [avg_cost, scale_infer]
C
choijulie 已提交
398 399


D
daminglu 已提交
400 401
def optimizer_func():
    return fluid.optimizer.SGD(learning_rate=0.2)
C
choijulie 已提交
402 403 404
```


D
daminglu 已提交
405
## 训练模型
C
choijulie 已提交
406

D
daminglu 已提交
407 408
### 定义训练环境
定义您的训练环境,可以指定训练是发生在CPU还是GPU上。
C
choijulie 已提交
409

D
daminglu 已提交
410 411 412 413 414 415 416
```python
use_cuda = False
place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace()
```

### 定义数据提供器
下一步是为训练和测试定义数据提供器。提供器读入一个大小为 `BATCH_SIZE`的数据。`paddle.dataset.movielens.train` 每次会在乱序化后提供一个大小为`BATCH_SIZE`的数据,乱序化的大小为缓存大小`buf_size`
C
choijulie 已提交
417

D
daminglu 已提交
418
```python
H
hutuxian 已提交
419 420
train_reader = fluid.io.batch(
    fluid.io.shuffle(
D
daminglu 已提交
421 422
        paddle.dataset.movielens.train(), buf_size=8192),
    batch_size=BATCH_SIZE)
C
choijulie 已提交
423

H
hutuxian 已提交
424
test_reader = fluid.io.batch(
D
daminglu 已提交
425 426 427
    paddle.dataset.movielens.test(), batch_size=BATCH_SIZE)
```

428 429
### 构造训练过程(trainer)
我们这里构造了一个训练过程,包括训练优化函数。
C
choijulie 已提交
430

D
daminglu 已提交
431 432 433
### 提供数据

`feed_order`用来定义每条产生的数据和`paddle.layer.data`之间的映射关系。比如,`movielens.train`产生的第一列的数据对应的是`user_id`这个特征。
C
choijulie 已提交
434 435

```python
D
daminglu 已提交
436 437 438 439
feed_order = [
    'user_id', 'gender_id', 'age_id', 'job_id', 'movie_id', 'category_id',
    'movie_title', 'score'
]
C
choijulie 已提交
440 441
```

442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462
### 构建训练程序以及测试程序
分别构建训练程序和测试程序,并引入训练优化器。

```python
main_program = fluid.default_main_program()
star_program = fluid.default_startup_program()
[avg_cost, scale_infer] = train_program()

test_program = main_program.clone(for_test=True)
sgd_optimizer = optimizer_func()
sgd_optimizer.minimize(avg_cost)
exe = fluid.Executor(place)

def train_test(program, reader):
    count = 0
    feed_var_list = [
        program.global_block().var(var_name) for var_name in feed_order
    ]
    feeder_test = fluid.DataFeeder(
    feed_list=feed_var_list, place=place)
    test_exe = fluid.Executor(place)
463
    accumulated = 0
464 465 466
    for test_data in reader():
        avg_cost_np = test_exe.run(program=program,
                                               feed=feeder_test.feed(test_data),
467 468
                                               fetch_list=[avg_cost])
        accumulated += avg_cost_np[0]
469
        count += 1
470
    return accumulated / count
471 472 473 474
```

### 构建训练主循环并开始训练
我们根据上面定义的训练循环数(`PASS_NUM`)和一些别的参数,来进行训练循环,并且每次循环都进行一次测试,当测试结果足够好时退出训练并保存训练好的参数。
D
daminglu 已提交
475

C
choijulie 已提交
476
```python
D
daminglu 已提交
477 478
# Specify the directory path to save the parameters
params_dirname = "recommender_system.inference.model"
479

480
from paddle.utils.plot import Ploter
481
train_prompt = "Train cost"
482
test_prompt = "Test cost"
483 484

plot_cost = Ploter(train_prompt, test_prompt)
485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501

def train_loop():
    feed_list = [
        main_program.global_block().var(var_name) for var_name in feed_order
    ]
    feeder = fluid.DataFeeder(feed_list, place)
    exe.run(star_program)

    for pass_id in range(PASS_NUM):
        for batch_id, data in enumerate(train_reader()):
            # train a mini-batch
            outs = exe.run(program=main_program,
                               feed=feeder.feed(data),
                               fetch_list=[avg_cost])
            out = np.array(outs[0])

            # get test avg_cost
502 503 504 505
            test_avg_cost = train_test(test_program, test_reader)

            plot_cost.append(train_prompt, batch_id, outs[0])
            plot_cost.append(test_prompt, batch_id, test_avg_cost)
506 507 508 509 510 511 512 513 514
            plot_cost.plot()

            if batch_id == 20:
                if params_dirname is not None:
                    fluid.io.save_inference_model(params_dirname, [
                                "user_id", "gender_id", "age_id", "job_id",
                                "movie_id", "category_id", "movie_title"
                        ], [scale_infer], exe)
                return
515 516
            print('EpochID {0}, BatchID {1}, Test Loss {2:0.2}'.format(
                            pass_id + 1, batch_id + 1, float(test_avg_cost)))
517 518

            if math.isnan(float(out[0])):
519
                sys.exit("got NaN loss, training failed.")
C
choijulie 已提交
520
```
521
开始训练
C
choijulie 已提交
522
```python
523
train_loop()
C
choijulie 已提交
524 525 526 527
```

## 应用模型

528
### 生成测试数据
D
daminglu 已提交
529
使用 create_lod_tensor(data, lod, place) 的API来生成细节层次的张量。`data`是一个序列,每个元素是一个索引号的序列。`lod`是细节层次的信息,对应于`data`。比如,data = [[10, 2, 3], [2, 3]] 意味着它包含两个序列,长度分别是3和2。于是相应地 lod = [[3, 2]],它表明其包含一层细节信息,意味着 `data` 有两个序列,长度分别是3和2。
C
choijulie 已提交
530

D
daminglu 已提交
531
在这个预测例子中,我们试着预测用户ID为1的用户对于电影'Hunchback of Notre Dame'的评分
C
choijulie 已提交
532

D
daminglu 已提交
533 534 535
```python
infer_movie_id = 783
infer_movie_name = paddle.dataset.movielens.movie_info()[infer_movie_id].title
H
hutuxian 已提交
536 537 538 539 540 541 542
user_id = np.array([1]).astype("int64").reshape(-1)
gender_id = np.array([1]).astype("int64").reshape(-1)
age_id = np.array([0]).astype("int64").reshape(-1)
job_id = np.array([10]).astype("int64").reshape(-1)
movie_id = np.array([783]).astype("int64").reshape(-1) # Hunchback of Notre Dame
category_id = fluid.create_lod_tensor(np.array([10, 8, 9], dtype='int64'), [[3]], place) # Animation, Children's, Musical
movie_title = fluid.create_lod_tensor(np.array([1069, 4140, 2923, 710, 988], dtype='int64'), [[5]],
D
daminglu 已提交
543
                                      place) # 'hunchback','of','notre','dame','the'
C
choijulie 已提交
544 545
```

546 547 548 549 550 551 552 553 554 555
### 构建预测过程并测试
与训练过程类似,我们需要构建一个预测过程。其中, `params_dirname`是之前用来存放训练过程中的各个参数的地址。

```python
place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace()
exe = fluid.Executor(place)

inference_scope = fluid.core.Scope()
```

D
daminglu 已提交
556 557
### 测试
现在我们可以进行预测了。我们要提供的`feed_order`应该和训练过程一致。
C
choijulie 已提交
558 559


D
daminglu 已提交
560
```python
561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581
with fluid.scope_guard(inference_scope):
    [inferencer, feed_target_names,
    fetch_targets] = fluid.io.load_inference_model(params_dirname, exe)

    results = exe.run(inferencer,
                          feed={
                               'user_id': user_id,
                              'gender_id': gender_id,
                              'age_id': age_id,
                              'job_id': job_id,
                              'movie_id': movie_id,
                              'category_id': category_id,
                              'movie_title': movie_title
                          },
                          fetch_list=fetch_targets,
                          return_numpy=False)
    predict_rating = np.array(results[0])
    print("Predict Rating of user id 1 on movie \"" + infer_movie_name +
              "\" is " + str(predict_rating[0][0]))
    print("Actual Rating of user id 1 on movie \"" + infer_movie_name +
              "\" is 4.")
D
daminglu 已提交
582
```
C
choijulie 已提交
583 584 585

## 总结

586
本章介绍了传统的个性化推荐系统方法和YouTube的深度神经网络个性化推荐系统,并以电影推荐为例,使用PaddlePaddle训练了一个个性化推荐神经网络模型。个性化推荐系统几乎涵盖了电商系统、社交网络、广告推荐、搜索引擎等领域的方方面面,而在图像处理、自然语言处理等领域已经发挥重要作用的深度学习技术,也将会在个性化推荐系统领域大放异彩。
C
choijulie 已提交
587

T
tink2123 已提交
588
<a name="参考文献"></a>
C
choijulie 已提交
589 590
## 参考文献

591 592 593 594
1. P. Resnick, N. Iacovou, etc. “[GroupLens: An Open Architecture for Collaborative Filtering of Netnews](http://ccs.mit.edu/papers/CCSWP165.html)”, Proceedings of ACM Conference on Computer Supported Cooperative Work, CSCW 1994. pp.175-186.
2. Sarwar, Badrul, et al. "[Item-based collaborative filtering recommendation algorithms.](http://files.grouplens.org/papers/www10_sarwar.pdf)*Proceedings of the 10th international conference on World Wide Web*. ACM, 2001.
3. Kautz, Henry, Bart Selman, and Mehul Shah. "[Referral Web: combining social networks and collaborative filtering.](http://www.cs.cornell.edu/selman/papers/pdf/97.cacm.refweb.pdf)" Communications of the ACM 40.3 (1997): 63-65. APA
4. [Peter Brusilovsky](https://en.wikipedia.org/wiki/Peter_Brusilovsky) (2007). *The Adaptive Web*. p. 325.
X
xiaoting 已提交
595
5. Robin Burke , [Hybrid Web Recommender Systems](http://citeseerx.ist.psu.edu/viewdoc/download?doi=10.1.1.435.7538&rep=rep1&type=pdf), pp. 377-408, The Adaptive Web, Peter Brusilovsky, Alfred Kobsa, Wolfgang Nejdl (Ed.), Lecture Notes in Computer Science, Springer-Verlag, Berlin, Germany, Lecture Notes in Computer Science, Vol. 4321, May 2007, 978-3-540-72078-2.
C
choijulie 已提交
596 597 598 599 600
6. Yuan, Jianbo, et al. ["Solving Cold-Start Problem in Large-scale Recommendation Engines: A Deep Learning Approach."](https://arxiv.org/pdf/1611.05480v1.pdf) *arXiv preprint arXiv:1611.05480* (2016).
7. Covington P, Adams J, Sargin E. [Deep neural networks for youtube recommendations](https://static.googleusercontent.com/media/research.google.com/zh-CN//pubs/archive/45530.pdf)[C]//Proceedings of the 10th ACM Conference on Recommender Systems. ACM, 2016: 191-198.


<br/>
X
xiaoting 已提交
601
<a rel="license" href="http://creativecommons.org/licenses/by-sa/4.0/"><img alt="知识共享许可协议" style="border-width:0" src="https://paddlepaddleimage.cdn.bcebos.com/bookimage/camo.png" /></a><br /><span xmlns:dct="http://purl.org/dc/terms/" href="http://purl.org/dc/dcmitype/Text" property="dct:title" rel="dct:type">本教程</span><a xmlns:cc="http://creativecommons.org/ns#" href="http://book.paddlepaddle.org" property="cc:attributionName" rel="cc:attributionURL">PaddlePaddle</a> 创作,采用 <a rel="license" href="http://creativecommons.org/licenses/by-sa/4.0/">知识共享 署名-相同方式共享 4.0 国际 许可协议</a>进行许可。