图神经网络实战——基于Node2Vec的电影推荐系统
0. 前言
Node2Vec 是一种用于获取节点嵌入 (node embedding) 的算法，它通过将图形数据转换为低维向量空间中的连续向量来捕捉节点之间的相似性。基于 Node2Vec 的电影推荐系统结合了图论和深度学习的方法来进行推荐。
在电影推荐系统中，可以将每部电影视为一个节点，而节点之间的连接则表示电影之间的关联或相似性。Node2Vec 算法可以在电影图谱上学习出每部电影的向量表示，这些向量可以捕获电影之间的隐含关系，例如共同的演员、导演、类型等等。一旦得到了电影的向量表示，就可以使用这些表示来计算电影之间的相似度，并基于相似度来进行推荐。通过将用户喜欢的电影与其向量表示进行比较，系统可以推荐与之相似的其他电影。在本节中，构建基于 Node2Vec 的电影推荐系统 (recommender system, RecSys)。
1. 基于 Node2Vec 的电影推荐系统
电影推荐系统 (recommender system, RecSys) 是图神经网络 (Graph Neural Networks, GNN) 最受欢迎的应用之一。我们知道 Word2Vec (以及 DeepWalk 和 Node2Vec) 的目标是生成能够测量相似度的向量。将电影(而非单词)进行编码，就能获取与给定输入标题最相似的电影，这就是 RecSys 的基本思想。
但问题在于如何对电影进行编码，我们首先想到可以创建电影的(有偏)随机游走，但这需要一个相似电影相互连接的图数据集，获取这类数据集并不容易。
另一种方法是根据用户评分来构建图。有多种基于评分构建图的技术，包括二分图、基于点互信息的边等。在本节中，我们将采用一种简单直观的方法，将受到同一用户喜爱的电影连接起来。然后，将使用所构建的图基于 Node2Vec 学习电影嵌入。
2. 构建训练数据集
(1) 首先，下载并解压电影数据集。在本节中，使用 MovieLens 数据集的子集，其中包含 100836 条评分、9742 部电影和 610 个用户。我们主要使用文件 atings.csv 和 movies.csv，第一个文件存储用户的所有评分，第二个文件可以用于将电影标识符转换为标题。
(2) 使用 pd.read_csv() 导入文件 atings.csv：
import pandas as pd
ratings = pd.read_csv('ml-100k/u.data', sep='\t', names=['user_id', 'movie_id', 'rating', 'unix_timestamp'])
print(ratings)
输出结果如下所示：
user_id movie_id rating unix_timestamp
0 196 242 3 881250949
1 186 302 3 891717742
2 22 377 1 878887116
3 244 51 2 880606923
4 166 346 1 886397596
... ... ... ... ...
99995 880 476 3 880175444
99996 716 204 5 879795543
99997 276 1090 1 874795795
99998 13 225 2 882399156
99999 12 203 3 879959583
[100000 rows x 4 columns]
(3) 导入 movies.csv：
movies = pd.read_csv('ml-100k/u.item', sep='|', usecols=range(2), names=['movie_id', 'title'], encoding='latin-1')
print(movies)
数据集输出结果如下所示：
movie_id title
0 1 Toy Story (1995)
1 2 GoldenEye (1995)
2 3 Four Rooms (1995)
3 4 Get Shorty (1995)
4 5 Copycat (1995)
... ... ...
1677 1678 Mat' i syn (1997)
1678 1679 B. Monkey (1998)
1679 1680 Sliding Doors (1998)
1680 1681 You So Crazy (1994)
1681 1682 Scream of Stone (Schrei aus Stein) (1991)
[1682 rows x 2 columns]
(4) 由于我们想得用户可能喜欢的电影，因此意味着评分为 1、2 和 3 并不非常重要，我们可以舍弃这些评分，只保留评分为 4 和 5 的数据：
ratings = ratings[ratings.rating >= 4]
print(ratings)
得到输出结果如下所示：
user_id movie_id rating unix_timestamp
5 298 474 4 884182806
7 253 465 5 891628467
11 286 1014 5 879781125
12 200 222 5 876042340
16 122 387 5 879270459
... ... ... ... ...
99988 421 498 4 892241344
99989 495 1091 4 888637503
99990 806 421 4 882388897
99991 676 538 4 892685437
99996 716 204 5 879795543
[55375 rows × 4 columns]
(5) 我们已经获取了由 610 个用户给出的 48580 条评分。下一步是统计两部电影被同一用户喜欢的次数，对数据集中的每个用户重复这一过程。为了简化操作，使用 defaultdict 数据结构，它可以自动创建缺失条目，而不会引发错误，使用这种结构统计同时被喜欢的电影的次数：
from collections import defaultdict
pairs = defaultdict(int)
(6) 循环浏览数据集中的用户列表：
for group in ratings.groupby("user_id"):
(7) 获取当前用户已赞过的电影列表：
user_movies = list(group[1]["movie_id"])
(8) 每当一对电影在同一个列表中同时出现时，就递增一个特定的计数器：
for i in range(len(user_movies)):
for j in range(i+1, len(user_movies)):
pairs[(user_movies[i], user_movies[j])] += 1
(9) pairs 对象存储了两部电影被同一用户点赞的次数，可以利用这些信息来构建图中的边。使用 networkx 库创建图 G：
G = nx.Graph()
(10) 对于 pairs 中的每对电影，解包这两部电影及其相应的评分：
for pair in pairs:
movie1, movie2 = pair
score = pairs[pair]
(11) 如果得分高于 20，就会在图中添加一个加权连接，根据得分将两部电影连接起来。忽略低于 20 的分数，因为这样会创建一个较大的图，而其中的连接并无太大意义：
if score >= 20:
G.add_edge(movie1, movie2, weight=score)
(12) 最终得到的图有 410 个节点(电影)和 14,936 条边，接下来训练 Node2Vec 学习节点嵌入：
print("Total number of graph nodes:", G.number_of_nodes())
print("Total number of graph edges:", G.number_of_edges())
# Total number of graph nodes: 410
# Total number of graph edges: 14936
3. 实现基于 Node2Vec 的电影推荐系统
我们可以重复使用 Node2Vec 一节中的 Node2Vec 实现，但在 Python 中有一个专门用于 Node2Vec 的库，在本节中，我们将使用此库实现 Node2Vec。
(1) 首先，使用 pip 命令安装 node2vec 库：
pip install node2vec
(2) 在代码中，导入 Node2Vec 类，并创建一个 Node2Vec 实例，根据 p 和 q 参数自动生成有偏随机游走序列：
from node2vec import Node2Vec
node2vec = Node2Vec(G, dimensions=64, walk_length=20, num_walks=200, p=2, q=1, workers=1)
(3) 窗口大小设置为 10 (前 5 个节点，后 5 个节点)，对这些有偏随机游走序列进行模型训练：
model = node2vec.fit(window=10, min_count=1, batch_words=4)
Node2Vec 模型完成训练后，可以像使用 gensim 库中的 Word2Vec 对象一样使用它。接下来，创建函数根据给定的标题推荐电影。
(4) 创建 recommend() 函数，将电影标题作为输入。首先，将标题转换为电影 ID：
def recommend(movie):
movie_id = str(movies[movies.title == movie].movie_id.values[0])
(5) 循环遍历五个最相似的单词向量。将这些 ID 转换成电影标题，并打印出相应的相似度分数：
for id in model.wv.most_similar(movie_id)[:5]:
title = movies[movies.movie_id == int(id[0])].title.values[0]
print(f'{title}: {id[1]:.2f}')
(6) 调用该函数来获得与《Star Wars (1977)》相似度最高的五部电影的余弦相似度：
recommend('Star Wars (1977)')
得到的输出结果如下所示：
Return of the Jedi (1983): 0.62
Raiders of the Lost Ark (1981): 0.57
Monty Python and the Holy Grail (1974): 0.53
Toy Story (1995): 0.47
Silence of the Lambs, The (1991): 0.46
可以看到，《Return of the Jedi》和《Raiders of the Lost Ark》与《Monty Python and the Holy Grail》最为相似，但得分相对较低( < 0.7)。尽管如此，对于我们初次构建推荐系统模型而言，这是一个相当不错的结果。在后续学习中，我们将使用更强大的模型和方法来构建更优秀的推荐系统模型。
相关链接
图神经网络实战（1）——图神经网络(Graph Neural Networks, GNN)基础
图神经网络实战（2）——图论基础
图神经网络实战（3）——基于DeepWalk创建节点表示
图神经网络实战（4）——基于Node2Vec改进嵌入质量
2911
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
