Press "Enter" to skip to content

超火的个性化推荐你再不会就OUT啦,让飞桨手把手来教你

导读:随着电子商务规模的不断扩大,电商平台的商品数量和种类呈爆发式增长,用户往往需要花费大量的时间才能找到自己想买的商品,这就是信息超载问题。为了解决这个难题,“个性化推荐”技术应运而生,有效地节约用户时间,提升电商成单率。本篇文章中,将为大家介绍个性化推荐系统的实现方法,并送上一份基于飞桨(PaddlePaddle)实现个性化推荐的代码教程。

 

1.  个性化推荐概述

 

日常生活中,当你打开某电商购物APP后,可能会遇到以下情形:

 

为什幺我和好友是同时打开的APP,但两个人的首页推荐商品不一致?

 

为什幺同样是搜索“T恤”,我和好友竟然出现了不一样的商品列表?

 

为什幺在我浏览了某品牌的运动鞋之后,“猜你喜欢”模块的商品列表变了?

 

为什幺对比好友的APP界面,我更喜欢我自己界面中的商品呢?

 

这就是个性化推荐在电商应用的例子。

 

个性化推荐,是指通过分析、挖掘用户行为,发现用户的个性化需求与兴趣特点,将用户可能感兴趣的信息或商品推荐给用户。

 

与搜索引擎不同,个性化推荐系统不需要用户准确地描述出自己的需求,而是根据用户的历史行为进行建模,主动提供满足用户兴趣和需求的信息。

 

个性化推荐几乎涵盖了电商系统、社交网络、广告推荐、搜索引擎等应用的方方面面。

 

个性化推荐技术的发展,可以分为传统推荐方法和深度学习推荐方法两个阶段。

 

传统推荐方法

 

1994年明尼苏达大学推出了GroupLens系统,这被认为是个性化推荐系统成为一个相对独立的研究方向的标志。GroupLens系统首次提出了基于协同过滤来完成推荐任务的思想,此后,基于该模型的协同过滤推荐算法引领了个性化推荐系统十几年的发展方向。

 

传统的个性化推荐系统方法主要有:

 

协同过滤推荐(Collaborative Filtering Recommendation):该方法是应用最广泛的技术之一,需要收集和分析用户的历史行为、活动和偏好。它通常可以分为两个子类:基于用户(User-Based)的推荐和基于物品(Item-Based)的推荐。该方法的一个关键优势是它不依赖于机器去分析物品的内容特征,因此它无需理解物品本身也能够准确地推荐诸如电影之类的复杂物品;缺点是对于没有任何行为的新用户存在冷启动的问题,同时也存在用户与商品之间的交互数据不够多造成的稀疏问题。值得一提的是,社交网络或地理位置等上下文信息都可以结合到协同过滤中去。

 

基于内容过滤推荐(Content-based Filtering Recommendation):该方法利用商品的内容描述,抽象出有意义的特征,通过计算用户的兴趣和商品描述之间的相似度,来给用户做推荐。优点是简单直接,不需要依据其他用户对商品的评价,而是通过商品属性进行商品相似度度量,从而推荐给用户所感兴趣商品的相似商品;缺点是对于没有任何行为的新用户同样存在冷启动的问题。

 

组合推荐(Hybrid Recommendation):运用不同的输入和技术共同进行推荐,以弥补各自推荐技术的缺点。

 

深度学习推荐方法

 

近些年来,深度学习在很多领域都取得了巨大的成功。学术界和工业界都在尝试将深度学习应用于个性化推荐系统领域中。深度学习具有优秀的自动提取特征的能力,能够学习多层次的抽象特征表示,并对异质或跨域的内容信息进行学习,可以一定程度上处理个性化推荐系统冷启动问题。

 

下面就为大家介绍个性化推荐的深度学习模型,以及如何使用PaddlePaddle实现该模型。

 

2.  效果展示

 

在正式开始之前,我们先看一下模型的最终效果:

 

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

 

<code>Input movie_id: 1962</code><code>Input user_id: 1</code><code>Prediction Score is 4.25</code>

 

3.  模型概览

 

我们将首先介绍YouTube的视频个性化推荐系统,然后再介绍融合推荐模型。

 

3.1.YouTube的深度神经网络个性化推荐系统

 

YouTube是世界上最大的视频上传、分享和发现网站,YouTube个性化推荐系统为超过10亿用户从不断增长的视频库中推荐个性化的内容。整个系统由两个神经网络组成:候选生成网络和排序网络。

 

候选生成网络从百万量级的视频库中生成上百个候选。

 

排序网络对候选进行打分排序,输出排名最高的数十个结果。

 

系统的整体结构如下:

 

 

3.1.1. 候选生成网络(Candidate Generation Network)

 

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

 

首先,将观看历史及搜索词记录这类历史信息,映射为向量后取平均值得到定长表示;同时,输入人口学特征以优化新用户的推荐效果,并将二值特征和连续特征归一化处理到[0, 1]范围。

 

接下来,将所有特征表示拼接为一个向量,并输入给非线形多层感知器(MLP)处理。

 

最后,训练时将MLP的输出给softmax做分类,预测时计算用户的综合特征(MLP的输出)与所有视频的相似度,取得分最高的k个作为候选生成网络的筛选结果。

 

下图是候选生成网络的结构:

 

 

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

 

 

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

 

考虑到softmax分类的类别数非常多,为了保证一定的计算效率,我们采取如下方式:

 

训练阶段,使用负样本类别采样将实际计算的类别数缩小至数千。

 

推荐(预测)阶段,忽略softmax的归一化计算(不影响结果),将类别打分问题简化为点积(dot product)空间中的最近邻(nearest neighbor)搜索问题,取与u最近的k个视频作为生成的候选。

 

3.1.2. 排序网络(Ranking Network)

 

排序网络的结构与候选生成网络类似,但是它的目标是对候选进行更细致的打分排序。

 

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

 

3.2.融合推荐模型

 

下面依次为大家介绍文本卷积神经网络和融合推荐模型。

 

3.2.1. 卷积神经网络(CNN)

 

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

 

卷积神经网络主要由卷积(convolution)和池化(pooling)操作构成,其应用及组合方式灵活多变,种类繁多。

 

我们将重点讲解下图所示的卷积神经网络文本分类模型:

 

假设待处理句子的长度为n,其中第i个词的词向量为 ,k为维度大小。

 

首先,进行词向量的拼接操作:将每h个词拼接起来形成一个大小为h的词窗口,记为 ,它表示词序列xi,xi+1,…,xi:xi+h的拼接,其中,i表示词窗口中第一个词在整个句子中的位置,取值范围从1到n-h+1,

 

其次,进行卷积操作:把卷积核(kernel) 应用于包含h各词的窗口 ,得到特征 ,其中 为偏置项(bias),f为非线性激活函数,如sigmoid。卷积核应用于句子中所有的词窗口 ,产生一个特征图(feature map):

 

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

 

3.2.2. 融合推荐模型

 

在融合推荐模型的电影个性化推荐系统中,主要分为以下步骤:

 

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

 

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

 

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

 

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

 

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

 

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

 

融合推荐模型的主要架构如下:

 

 

4.  飞桨实战

 

下面送上基于飞桨(PaddlePaddle)实现个性化推荐的代码教程。

 

本教程的源码和ipynb文件位于book/recommender_system,初次使用请先参考Book文档使用说明。

 

4.1.数据准备

 

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

 

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

 

<code>import paddle</code><code>movie_info = paddle.dataset.movielens.movie_info()</code><code>print movie_info.values()[0]</code><code> </code><code># Run this block to showdataset's documentation</code><code># help(paddle.dataset.movielens)</code>

 

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

 

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

 

<code>movie_info = paddle.dataset.movielens.movie_info()</code><code>print movie_info.values()[0]</code><code> </code><code><MovieInfo id(1), title(Toy Story ),categories(['Animation', "Children's", 'Comedy'])></code>

 

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

 

<code>user_info = paddle.dataset.movielens.user_info()</code><code>print user_info.values()[0]</code><code> </code><code><UserInfo id(1), gender(F), age(1), job(10)></code>

 

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

 

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

 

<code>"Under 18"</code><code>"18-24"</code><code>"25-34"</code><code>"35-44"</code><code>"45-49"</code><code>"50-55"</code><code>"56+"</code>

 

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

 

<code>"other" or not specified</code><code>"academic/educator"</code><code>"artist"</code><code>"clerical/admin"</code><code>"college/grad student"</code><code>"customer service"</code><code>"doctor/health care"</code><code>"executive/managerial"</code><code>"farmer"</code><code>"homemaker"</code><code>"K-12 student"</code><code>"lawyer"</code><code>"programmer"</code><code>"retired"</code><code>"sales/marketing"</code><code>"scientist"</code><code>"self-employed"</code><code>"technician/engineer"</code><code>"tradesman/craftsman"</code><code>"unemployed"</code><code>"writer"</code>

 

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

 

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

 

<code>train_set_creator = paddle.dataset.movielens.train()</code><code>train_sample = next(train_set_creator())</code><code>uid = train_sample[0]</code><code>mov_id = train_sample[len(user_info[uid].value())]</code><code>print "User %s rates Movie %swith Score %s"%(user_info[uid],movie_info[mov_id], train_sample[-1])</code><code>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]</code>

 

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

 

4.2.模型配置说明

 

下面我们开始根据输入数据的形式配置模型。首先引入所需的库函数以及定义全局变量。

 

IS_SPARSE: embedding中是否使用稀疏更新

 

PASS_NUM: epoch数量

 

<code>from __future__ import print_function</code><code>import math</code><code>import sys</code><code>import numpy as np</code><code>import paddle</code><code>import paddle.fluid as fluid</code><code>import paddle.fluid.layers as layers</code><code>import paddle.fluid.nets as nets</code><code> </code><code>IS_SPARSE = True</code><code>BATCH_SIZE = 256</code><code>PASS_NUM = 20</code>

 

然后为我们的用户特征综合模型定义模型配置:

 

<code>def get_usr_combined_features():</code><code>    """ network definition for user part """</code><code> </code><code>    USR_DICT_SIZE =paddle.dataset.movielens.max_user_id() + 1</code><code> </code><code>    uid = layers.data(name='user_id', shape=[1], dtype='int64')</code><code> </code><code>    usr_emb = layers.embedding(</code><code>        input=uid,</code><code>        dtype='float32',</code><code>        size=[USR_DICT_SIZE, 32],</code><code>        param_attr='user_table',</code><code>        is_sparse=IS_SPARSE)</code><code> </code><code>    usr_fc = layers.fc(input=usr_emb, size=32)</code><code> </code><code>    USR_GENDER_DICT_SIZE = 2</code><code> </code><code>    usr_gender_id = layers.data(name='gender_id', shape=[1], dtype='int64')</code><code> </code><code>    usr_gender_emb = layers.embedding(</code><code>        input=usr_gender_id,</code><code>        size=[USR_GENDER_DICT_SIZE, 16],</code><code>        param_attr='gender_table',</code><code>        is_sparse=IS_SPARSE)</code><code> </code><code>    usr_gender_fc =layers.fc(input=usr_gender_emb, size=16)</code><code> </code><code>    USR_AGE_DICT_SIZE =len(paddle.dataset.movielens.age_table)</code><code>    usr_age_id = layers.data(name='age_id', shape=[1], dtype="int64")</code><code> </code><code>    usr_age_emb = layers.embedding(</code><code>        input=usr_age_id,</code><code>        size=[USR_AGE_DICT_SIZE, 16],</code><code>        is_sparse=IS_SPARSE,</code><code>        param_attr='age_table')</code><code> </code><code>    usr_age_fc = layers.fc(input=usr_age_emb,size=16)</code><code> </code><code>    USR_JOB_DICT_SIZE =paddle.dataset.movielens.max_job_id() + 1</code><code>    usr_job_id = layers.data(name='job_id', shape=[1], dtype="int64")</code><code> </code><code>    usr_job_emb = layers.embedding(</code><code>        input=usr_job_id,</code><code>        size=[USR_JOB_DICT_SIZE, 16],</code><code>        param_attr='job_table',</code><code>        is_sparse=IS_SPARSE)</code><code> </code><code>    usr_job_fc = layers.fc(input=usr_job_emb,size=16)</code><code> </code><code>    concat_embed = layers.concat(</code><code>        input=[usr_fc, usr_gender_fc,usr_age_fc, usr_job_fc], axis=1)</code><code> </code><code>    usr_combined_features =layers.fc(input=concat_embed, size=200, act="tanh")</code><code> </code><code>    return usr_combined_features</code>

 

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

 

然后,我们对于所有的用户特征,均输入到一个全连接层(fc)中。将所有特征融合为一个200维度的特征。

 

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

 

<code>def get_mov_combined_features():</code><code>    """network definition for item(movie) part"""</code><code> </code><code>    MOV_DICT_SIZE =paddle.dataset.movielens.max_movie_id() + 1</code><code> </code><code>    mov_id = layers.data(name='movie_id', shape=[1], dtype='int64')</code><code> </code><code>    mov_emb = layers.embedding(</code><code>        input=mov_id,</code><code>        dtype='float32',</code><code>        size=[MOV_DICT_SIZE, 32],</code><code>        param_attr='movie_table',</code><code>        is_sparse=IS_SPARSE)</code><code> </code><code>    mov_fc = layers.fc(input=mov_emb, size=32)</code><code> </code><code>    CATEGORY_DICT_SIZE = len(paddle.dataset.movielens.movie_categories())</code><code> </code><code>    category_id = layers.data(</code><code>        name='category_id', shape=[1], dtype='int64', lod_level=1)</code><code> </code><code>    mov_categories_emb = layers.embedding(</code><code>        input=category_id,size=[CATEGORY_DICT_SIZE, 32], is_sparse=IS_SPARSE)</code><code> </code><code>    mov_categories_hidden =layers.sequence_pool(</code><code>        input=mov_categories_emb, pool_type="sum")</code><code> </code><code>    MOV_TITLE_DICT_SIZE =len(paddle.dataset.movielens.get_movie_title_dict())</code><code> </code><code>    mov_title_id = layers.data(</code><code>        name='movie_title', shape=[1], dtype='int64', lod_level=1)</code><code> </code><code>    mov_title_emb = layers.embedding(</code><code>        input=mov_title_id,size=[MOV_TITLE_DICT_SIZE, 32], is_sparse=IS_SPARSE)</code><code> </code><code>    mov_title_conv = nets.sequence_conv_pool(</code><code>        input=mov_title_emb,</code><code>        num_filters=32,</code><code>        filter_size=3,</code><code>        act="tanh",</code><code>        pool_type="sum")</code><code> </code><code>    concat_embed = layers.concat(</code><code>        input=[mov_fc, mov_categories_hidden,mov_title_conv], axis=1)</code><code> </code><code>    mov_combined_features =layers.fc(input=concat_embed, size=200, act="tanh")</code><code> </code><code>    return mov_combined_features</code>

 

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

 

最后,我们定义一个inference_program来使用余弦相似度计算用户特征与电影特征的相似性。

 

<code>def inference_program():</code><code>    """the combined network"""</code><code> </code><code>    usr_combined_features =get_usr_combined_features()</code><code>    mov_combined_features =get_mov_combined_features()</code><code> </code><code>    inference =layers.cos_sim(X=usr_combined_features, Y=mov_combined_features)</code><code>    scale_infer = layers.scale(x=inference,scale=5.0)</code><code> </code><code>    return scale_infer</code>

 

进而,我们定义一个train_program来使用inference_program计算出的结果,在标记数据的帮助下来计算误差。我们还定义了一个optimizer_func来定义优化器。

 

<code>def train_program():</code><code>    """define the cost function"""</code><code> </code><code>    scale_infer = inference_program()</code><code> </code><code>    label = layers.data(name='score', shape=[1], dtype='float32')</code><code>    square_cost =layers.square_error_cost(input=scale_infer, label=label)</code><code>    avg_cost = layers.mean(square_cost)</code><code> </code><code>    return [avg_cost,scale_infer]</code><code> </code><code>def optimizer_func():</code><code>    returnfluid.optimizer.SGD(learning_rate=0.2)</code>

 

4.3.模型训练

 

(1)定义训练环境

 

定义您的训练环境,可以指定训练是发生在CPU还是GPU上。

 

<code>use_cuda = False</code><code>place = fluid.CUDAPlace(0) if use_cuda elsefluid.CPUPlace()</code>

 

(2)定义数据提供器

 

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

 

<code>train_reader = paddle.batch(</code><code>    paddle.reader.shuffle(</code><code>        paddle.dataset.movielens.train(),buf_size=8192),</code><code>    batch_size=BATCH_SIZE)</code><code> </code><code>test_reader = paddle.batch(</code><code>    paddle.dataset.movielens.test(),batch_size=BATCH_SIZE)</code>

 

(3)构造训练过程

 

我们这里构造了一个训练过程,包括训练优化函数。

 

(4)提供数据

 

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

 

feed_order = [    'user_id', 'gender_id', 'age_id', 'job_id', 'movie_id', 'category_id',    'movie_title', 'score']

 

(5)构造训练程序以及测试程序

 

分别构建训练程序和测试程序,并引入训练优化器。

 

<code>main_program =fluid.default_main_program()</code><code>star_program =fluid.default_startup_program()</code><code>[avg_cost, scale_infer] =train_program()</code><code> </code><code>test_program =main_program.clone(for_test=True)</code><code>sgd_optimizer =optimizer_func()</code><code>sgd_optimizer.minimize(avg_cost)</code><code>exe = fluid.Executor(place)</code><code> </code><code>deftrain_test(program, reader):</code><code>    count = 0</code><code>    feed_var_list = [</code><code>        program.global_block().var(var_name) for var_name in feed_order</code><code>    ]</code><code>    feeder_test = fluid.DataFeeder(</code><code>    feed_list=feed_var_list, place=place)</code><code>    test_exe = fluid.Executor(place)</code><code>    accumulated = 0</code><code>    for test_data in reader():</code><code>        avg_cost_np =test_exe.run(program=program, </code><code>        feed=feeder_test.feed(test_data),</code><code>        fetch_list=[avg_cost])</code><code>        accumulated += avg_cost_np[0]</code><code>        count += 1</code><code>    return accumulated /count</code>

 

(6)构造训练主循环

 

我们根据上面定义的训练循环数(PASS_NUM)和一些别的参数,来进行训练循环,并且每次循环都进行一次测试,当测试结果足够好时退出训练并保存训练好的参数。

 

<code># Specify the directory pathto save the parameters</code><code>params_dirname = "recommender_system.inference.model"</code><code> </code><code>from paddle.utils.plot import Ploter</code><code>train_prompt = "Train cost"</code><code>test_prompt = "Test cost"</code><code> </code><code>plot_cost = Ploter(train_prompt,test_prompt)</code><code> </code><code>deftrain_loop():</code><code>    feed_list = [</code><code>       main_program.global_block().var(var_name) for var_name in feed_order</code><code>    ]</code><code>    feeder = fluid.DataFeeder(feed_list, place)</code><code>    exe.run(star_program)</code><code> </code><code>    for pass_id in range(PASS_NUM):</code><code>        for batch_id, datain enumerate(train_reader()):</code><code>            # train a mini-batch</code><code>            outs =exe.run(program=main_program,</code><code>                              feed=feeder.feed(data),</code><code>                               fetch_list=[avg_cost])</code><code>            out = np.array(outs[0])</code><code> </code><code>            # get test avg_cost</code><code>            test_avg_cost =train_test(test_program, test_reader)</code><code> </code><code>            plot_cost.append(train_prompt,batch_id, outs[0])</code><code>            plot_cost.append(test_prompt,batch_id, test_avg_cost)</code><code>            plot_cost.plot()</code><code> </code><code>            if batch_id == 20:</code><code>                if params_dirname isnotNone:</code><code>                   fluid.io.save_inference_model(params_dirname, [</code><code>                                "user_id", "gender_id", "age_id", "job_id",</code><code>                                "movie_id", "category_id", "movie_title"</code><code>                        ], [scale_infer], exe)</code><code>                return</code><code>            print('EpochID {0}, BatchID {1}, Test Loss {2:0.2}'.format(</code><code>                            pass_id + 1, batch_id + 1, float(test_avg_cost)))</code><code> </code><code>            if math.isnan(float(out[0])):</code><code>                sys.exit("got NaN loss, training failed.")</code>

 

(7)开始训练

 

train_loop()

 

4.4.模型测试

 

(1)生成测试数据

 

使用create_lod_tensor(data, lod, place) 的API来生成细节层次的张量。data是一个序列,每个元素是一个索引号的序列。lod是细节层次的信息,对应于data。比如,data = [[10, 2, 3], [2, 3]] 意味着它包含两个序列,长度分别是3和2。于是相应地 lod = [[3, 2]],它表明其包含一层细节信息,意味着 data 有两个序列,长度分别是3和2。

 

在这个预测例子中,我们试着预测用户ID为1的用户对于电影’Hunchback of Notre Dame’的评分。

 

<code>infer_movie_id = 783</code><code>infer_movie_name = paddle.dataset.movielens.movie_info()[infer_movie_id].title</code><code>user_id =fluid.create_lod_tensor([[np.int64(1)]], [[1]], place)</code><code>gender_id =fluid.create_lod_tensor([[np.int64(1)]], [[1]], place)</code><code>age_id =fluid.create_lod_tensor([[np.int64(0)]], [[1]], place)</code><code>job_id =fluid.create_lod_tensor([[np.int64(10)]], [[1]], place)</code><code>movie_id =fluid.create_lod_tensor([[np.int64(783)]], [[1]], place) # Hunchback ofNotre Dame</code><code>category_id =fluid.create_lod_tensor([np.array([10, 8, 9], dtype='int64')], [[3]], place) # Animation,Children's, Musical</code><code>movie_title =fluid.create_lod_tensor([np.array([1069, 4140, 2923, 710, 988], dtype='int64')], [[5]],</code><code>                                      place) # 'hunchback','of','notre','dame','the'</code>

 

(2)构建预测过程

 

与训练过程类似,我们需要构建一个预测过程。其中, params_dirname是之前用来存放训练过程中的各个参数的地址。

 

<code>place = fluid.CUDAPlace(0) if use_cuda elsefluid.CPUPlace()</code><code>exe = fluid.Executor(place)</code><code> </code><code>inference_scope =fluid.core.Scope()</code>

 

(3)进行测试

 

现在我们可以进行预测了。我们要提供的feed_order应该和训练过程保持一致。

 

<code>with fluid.scope_guard(inference_scope):</code><code>    [inferencer, feed_target_names,</code><code>    fetch_targets] =fluid.io.load_inference_model(params_dirname, exe)</code><code> </code><code>    results = exe.run(inferencer,</code><code>                          feed={</code><code>                               'user_id': user_id,</code><code>                              'gender_id': gender_id,</code><code>                              'age_id': age_id,</code><code>                              'job_id': job_id,</code><code>                              'movie_id': movie_id,</code><code>                              'category_id': category_id,</code><code>                              'movie_title': movie_title</code><code>                          },</code><code>                         fetch_list=fetch_targets,</code><code>                          return_numpy=False)</code><code>    predict_rating = np.array(results[0])</code><code>    print("Predict Rating of user id 1 on movie \"" + infer_movie_name +</code><code>              "\" is " +str(predict_rating[0][0]))</code><code>    print("Actual Rating of user id 1 on movie \"" + infer_movie_name +</code><code>              "\" is 4.") </code>

 

赶快动手尝试下吧!

 

想与更多的深度学习开发者交流,请加入飞桨官方QQ群:796771754。

 

如果您想详细了解更多飞桨PaddlePaddle的相关内容,请参阅以下文档。

 

官网地址:

 

https://www.paddlepaddle.org.cn/

 

项目地址:

 

https://github.com/PaddlePaddle/book/blob/develop/05.recommender_system/README.cn.md

Be First to Comment

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注