你好,我是王喆。

今天又是一堂实战课。在这节课里,我会带你利用我们现阶段掌握的所有知识,来实现SparrowRecSys中“猜你喜欢”的功能。具体来说,我们会根据一位用户的历史行为,为TA推荐可能喜欢的电影。这个功能几乎会用到所有的推荐系统模块,包括离线的特征工程、模型训练以及线上的模型服务和推荐逻辑的实现。

如果说完成了第14讲的“相似电影”功能,还只是你“武功小成”的标志,那啃透了这节课的实践,就代表你掌握了推荐系统技术框架中的大部分内容,你就能在推荐系统的实际工作中做到“驾轻就熟”啦。

“清点技能库”,看看我们已有的知识储备有哪些

正式开始实践之前,我们还是先来清点一次自己的技能库。看看经过推荐模型篇的学习,我们技能库中的“兵器”又增加了多少,哪些可以用来实现“猜你喜欢”这个功能。下面,我就按照从离线到线上,由数据到模型的顺序,为你依次梳理一下特征工程、模型离线训练、模型服务、推荐服务器逻辑这四大部分的技能点。

1. 模型特征工程

特征工程是所有机器学习项目的起点,咱们的推荐模型也不例外。为了训练推荐模型,我们需要准备好模型所需的样本和特征。此外,在进行模型线上推断的时候,推荐服务器也需要线上实时拼装好包含了用户特征、物品特征、场景特征的特征向量,发送给推荐模型进行实时推断。

在“模型实战准备二”这一讲,我们就通过Spark处理好了TensorFlow训练所需的训练样本,并把Spark处理好的特征插入了Redis特征数据库,供线上推断使用。不熟悉这部分内容的同学,最好再复习一下相关内容,把这把武器装进自己的技能库。

2. 模型离线训练

为了在线上做出尽量准确或者说推荐效果尽量好的排序,我们需要在离线训练好排序所用的推荐模型。

我们在这一篇中学习和实践的所有深度推荐模型,都是围绕着这个目的展开的。虽然这些深度推荐模型的结构各不相同,但它们的输入、输出都是一致的,输入是由不同特征组成的特征向量,输出是一个分数,这个分数的高低代表了这组特征对应的用户对物品的喜好程度。

具体实践的时候,我们在TensorFlow平台上实现了Embedding MLP、Wide&Deep、NeuralCF、双塔模型、DeepFM等几种不同的深度推荐模型,它们中的任何一个都可以支持“猜你喜欢”的排序功能。

在实际的工业级系统中,我们会通过离线、在线等不同的评估手段来挑出最好的那个模型,去支持真实的应用场景。在SparrowRecsys中,我们以NeuralCF模型为例,实现“猜你喜欢”功能。其他模型的上线方法与NeuralCF几乎一致,唯一的区别是,对于不同的模型来说,它们在模型服务的部分需要载入不同的模型文件,并且在线上预估的部分也要传入模型相应的输入特征。

3. 模型服务

模型服务是推荐系统中连接线上环境和线下环境的纽带之一(另一个关键的纽带是特征数据库)。

在离线训练好模型之后,为了让模型在线上发挥作用,做出实时的推荐排序,我们需要通过模型服务的模块把推荐模型部署上线。我们曾经在第13讲中详细介绍过主流的模型服务方法,它们是“预存推荐结果”“预训练Embedding+轻量级线上模型”“利用PMML转换和部署模型”以及“TensorFlow Serving”。因为我们这一篇的深度学习模型都是基于TensorFlow训练的,所以这节课我们也会采用TensorFlow Serving作为模型服务的方式。

4. 推荐服务器内部逻辑实现

模型服务虽然可以做到“猜你喜欢”中电影的排序,但要进行排序,仍然需要做大量的准备工作,比如候选集的获取,召回层的构建,特征的获取和拼装等等。这些推荐逻辑都是在推荐服务器内部实现的。推荐服务器就像推荐系统的线上的心脏,是所有线上模块的核心。

我们曾经在“相似电影”功能中实现过整套的推荐逻辑,今天我们重点关注其中不同的部分,就是特征的拼装,以及从推荐服务器内部请求模型服务API的方法。

至此,我们准备好了自己的技能库。接下来,就让我们使出十八般武艺,来打造“猜你喜欢”这个推荐功能吧。

“猜你喜欢”推荐功能的技术架构

与“相似电影”功能一样,“猜你喜欢”相关的技术架构同样是由数据模型部分、线上部分和前端部分组成的。我们先来看看整个功能的技术架构图,再来说说每部分的具体实现细节。下图1 就是“猜你喜欢”功能的技术架构图,接下来,你就跟着我,按照从左上到右下的顺序,一起随着数据流的走向过一遍这个架构图吧。

首先,我们来看数据和模型部分。左上角是我们使用的数据集MovieLens,它经过Spark的处理之后,会生成两部分数据,分别从两个出口出去,特征部分会存入Redis供线上推断时推荐服务器使用,样本部分则提供给TensorFlow训练模型。

TensorFlow完成模型训练之后,会导出模型文件,然后模型文件会载入到TensorFlow Serving中,接着TensorFlow Serving会对外开放模型服务API,供推荐服务器调用。

接下来,我们再看推荐服务器部分。在这部分里,基于MovieLens数据集生成的候选电影集合会依次经过候选物品获取、召回层、排序层这三步,最终生成“猜你喜欢”的电影推荐列表,然后返回给前端,前端利用HTML和JavaScript把它们展示给用户。

整个过程中,除了排序层和TensorFlow Serving的实现,其他部分我们都已经在之前的实战中一一实现过。所以今天,我们会重点讲解推荐服务器排序层和TensorFlow Serving的实现。

排序层+TensorFlow Serving的实现

在推荐服务器内部,经过召回层之后,我们会得到几百量级的候选物品集。最后我们到底从这几百部电影中推荐哪些给用户,这个工作就交由排序层来处理。因为排序的工作是整个推荐系统提高效果的重中之重,在业界的实际应用中,往往交由评估效果最好的深度推荐模型来处理。整个的排序过程可以分为三个部分:

  1. 准备线上推断所需的特征,拼接成JSON格式的特征样本;

  2. 把所有候选物品的特征样本批量发送给TensorFlow Serving API;

  3. 根据TensorFlow Serving API返回的推断得分进行排序,生成推荐列表。

接下来,我们就详细来讲讲这三步中的实现重点。

首先,第一步的实现重点在于特征样本的拼接。因为实践例子里,我们选用了NeuralCF作为排序模型,而NerualCF所需的特征只有 userIditemId ,所以特征是比较好准备的。我们下面看一下如何拼接特征形成模型推断所需的样本。详细的代码,你可以参考 com.wzhe.sparrowrecsys.online.recprocess.RecForYouProcess。

/**
 * call TenserFlow serving to get the NeuralCF model inference result
 * @param user              input user
 * @param candidates        candidate movies
 * @param candidateScoreMap save prediction score into the score map
 */
public static void callNeuralCFTFServing(User user, List<Movie> candidates, HashMap<Movie, Double> candidateScoreMap){
    if (null == user || null == candidates || candidates.size() == 0){
        return;
    }
    //保存所有样本的JSON数组
    JSONArray instances = new JSONArray();
    for (Movie m : candidates){
        JSONObject instance = new JSONObject();
        //为每个样本添加特征,userId和movieId
        instance.put("userId", user.getUserId());
        instance.put("movieId", m.getMovieId());
        instances.put(instance);
    }
    JSONObject instancesRoot = new JSONObject();
    instancesRoot.put("instances", instances);
    //请求TensorFlow Serving API
    String predictionScores = asyncSinglePostRequest("http://localhost:8501/v1/models/recmodel:predict", instancesRoot.toString());
    //获取返回预估值
    JSONObject predictionsObject = new JSONObject(predictionScores);
    JSONArray scores = predictionsObject.getJSONArray("predictions");
    //将预估值加入返回的map
    for (int i = 0 ; i < candidates.size(); i++){
        candidateScoreMap.put(candidates.get(i), scores.getJSONArray(i).getDouble(0));
    }
}

在代码中,我们先把 userIdmovieId 加入了JSON格式的样本中,然后再把样本加入到Json数组中。接下来,我们又以 http post 请求的形式把这些JSON样本发送给TensorFlow Serving的API,进行批量预估。在收到预估得分后,保存在候选集 map 中,供排序层进行排序。

第二步的重点在于如何建立起TensorFlow Serving API。事实上,我们通过第13讲模型服务的实践部分,已经能够搭建起一个测试模型的API了。

想要搭建起我们自己的TensorFlow Serving API,只需要把之前载入的测试模型文件换成我们自己的模型文件就可以了。这里,我就以NerualCF模型为例,带你看一看模型文件是如何被导出和导入的。

首先是模型的导出。在NeuralCF的TensorFlow实现中,我们已经把训练好的模型保存在了 model 这个结构中,接下来需要调用 tf.keras.models.save_model 这一函数来把模型序列化。

从下面的代码中你可以看到,这一函数需要传入的参数有要保存的 model 结构,保存的路径,还有是否覆盖路径 overwrite 等等。其中,我们要注意的是保存路径。你可以看到,我在保存路径中加上了一个模型版本号002,这对于TensorFlow Serving是很重要的,因为TensorFlow Serving总是会找到版本号最大的模型文件进行载入,这样做就保证了我们每次载入的都是最新训练的模型。详细代码请你参考 NeuralCF.py。

tf.keras.models.save_model(
    model,
  "file:///Users/zhewang/Workspace/SparrowRecSys/src/main/resources/webroot/modeldata/neuralcf/002",
    overwrite=True,
    include_optimizer=True,
    save_format=None,
    signatures=None,
    options=None
)

其次是模型的导入,导入命令非常简单就是TensorFlow Serving API的启动命令,我们直接看下面命令中的参数。

docker run -t --rm -p 8501:8501     -v "/Users/zhewang/Workspace/SparrowRecSys/src/main/resources/webroot/modeldata/neuralcf:/models/recmodel"     -e MODEL_NAME=recmodel     tensorflow/serving &

这里面最重要的参数,就是指定载入模型的路径和预估url,而载入路径就是我们刚才保存模型的路径:/Users/zhewang/Workspace/SparrowRecSys/src/main/resources/webroot/modeldata/neuralcf。但是在这里,我们没有加模型的版本号。这是为什么呢?因为版本号是供TensorFlow Serving查找最新模型用的,TensorFlow Serving在模型路径上会自动找到版本号最大的模型载入,因此不可以在载入路径上再加上版本号。

除此之外,冒号后的部分“/models/recmodel”指的是TensorFlow Serving API在这个模型上的具体url,刚才我们是通过请求http://localhost:8501/v1/models/recmodel:predict 获取模型预估值的,请求连接中的models/recmodel就是在这里设定的。

在正确执行上面的命令后,我们就可以在Docker上运行起TensorFlow Serving的API了。

最后,我们来看第三步的实现重点:获取返回得分和排序。 我们先来看一下TensorFlow Serving API的返回得分格式。它的返回值也是一个JSON 数组的格式,数组中每一项对应着之前发送过去的候选电影样本,所以我们只要把返回的预估值赋给相应的样本,然后按照预估值排序就可以了。详细的过程你也可以参考com.wzhe.sparrowrecsys.online.recprocess.RecForYouProcess中全部排序层的代码。

{
    "predictions": [[0.824034274], [0.86393261], [0.921346784], [0.957705915], [0.875154734], [0.905113697], [0.831545711], [0.926080644], [0.898158073]...
    ]
}

如果你已经正确建立起了Redis和TensorFlow Serving API服务,并且已经分别导入了特征数据和模型文件,我们就可以启动Sparrow Recsys Server,查看“猜你喜欢”的结果了。图2是用户ID为6的用户在NerualCF模型下的推荐结果,注意通过在连接中设置model变量为nerualcf,来决定产生结果的模型。

通过用户的评分历史(User Watched Movies)我们可以看到该用户偏向于观看动作类的电影,同时夹杂着一些爱情片和动画片,而在下方的“猜你喜欢”(Recommended For You)的结果中,我们也可以看到Sparrow Recsys为他推荐的电影也包含了这三类电影。有兴趣的话,你可以多在Sparrow Recsys里面逛一逛,看看推荐的结果在你眼中是不是合理。

小结

今天我们通过实现“猜你喜欢”功能串联起了我们之前所有学过的知识。希望在你看到推荐结果的时候,有种“武功大成,驾轻就熟”的感觉。要知道,这里面所有的代码都是你曾经学习过的,这里面每个结果都是你通过自己的所学生成的。希望你能在这里为自己鼓掌,这是一个不小的里程碑。

下面我们再重点总结一下今天实践用到的技术。首先,我们利用Spark对MovieLens原始数据进行了处理,生成了训练样本和特征,样本供TensorFlow进行模型训练,特征存入Redis供线上推断使用。

在TensorFlow平台上,我们以NeuralCF模型为例,训练并导出了NeuralCF的模型文件。然后使用TensorFlow Serving载入模型文件,建立线上模型服务API。推荐服务器的排序层从Redis中取出用户特征和物品特征,组装好JSON格式的特征数据,发送给TensorFlow Serving API,再根据返回的预估分数进行排序,最终生成“猜你喜欢”的推荐列表。

虽然我们实现了猜你喜欢的功能,但是课程进行到这里你一定会有一个疑问:我们的推荐结果到底是好还是坏呢?我们总不能总是人肉去查看结果好坏吧,这样效率又低,又不准确。没错,推荐系统的效果评估是有一套非常完整的评估体系的,别着急,从下一篇的模型评估篇开始,我们就会系统性地讲解推荐系统的评估方法,期待继续与你同行。

课后思考

推荐系统的特征预处理是一项很重要的工作,比如一些连续特征的归一化,分桶等等。那么这些预处理的过程,我们应该放在线上部分的哪里完成呢?是在Tensorflow Serving的部分,还是在推荐服务器内部,还是在离线部分完成?你有什么好的想法吗?

期待在留言区看到你的想法和思考,我们下一节课见!