如何应用node2vec建立链路预测模型

Question

我一直在尝试学习python编程，我正在尝试实现一个链接预测项目。

我有一个包含元组对的列表，如：

ten_author_pairs = [('creutzig', 'gao'), ('creutzig', 'linshaw'), ('gao', 'linshaw'), ('jing', 'zhang'), ('jing', 'liu'), ('zhang', 'liu'), ('jing', 'xu'), ('briant', 'einav'), ('chen', 'gao'), ('chen', 'jing'), ('chen', 'tan')]

我能够使用以下代码生成未连接的对，即原始列表中不存在的对：

#generating negative examples - 
from itertools import combinations
elements = list(set([e for l in ten_author_pairs for e in l])) # find all unique elements
complete_list = list(combinations(elements, 2)) # generate all possible combinations
#convert to sets to negate the order
set1 = [set(l) for l in ten_author_pairs]
complete_set = [set(l) for l in complete_list]
# find sets in `complete_set` but not in `set1`
ten_unconnnected = [list(l) for l in complete_set if l not in set1]
print(len(ten_author_pairs))
print(len(ten_unconnnected))

这导致我有一个非常不平衡的数据集——这可能是现实生活数据集的预期情况。

接下来，为了应用node2vec，首先，我将这两个列表转换为数据帧-

df = pd.DataFrame(ten_author_pairs, columns = ['u1','u2'])
df_negative = pd.DataFrame(ten_unconnected, columns = ['u1','u2'])
df['link'] = 1 #for connected pairs
df_negative['link'] = 0 #for unconnected pairs
df_new = pd.concat([df,df_negative])

然后，我制作图并应用node2vec如下：

# build graph
G_data = nx.from_pandas_edgelist(df_new, "u1", "u2", create_using=nx.Graph())
#!pip install node2vec
from node2vec import Node2Vec
# Generate walks
node2vec = Node2Vec(G_data, dimensions=100, walk_length=16, num_walks=50)
# train node2vec model
n2w_model = node2vec.fit(window=7, min_count=1)

最后，我使用Logistic回归进行链接预测，如下所示：

x = [(n2w_model[str(i)]+n2w_model[str(j)]) for i,j in zip(df_new['u1'], df_new['u2'])]
from sklearn.model_selection import train_test_split
xtrain, xtest, ytrain, ytest = train_test_split(np.array(x), df_new['link'], 
test_size = 0.3, 
random_state = 35)
from sklearn.linear_model import LogisticRegression
lr = LogisticRegression(class_weight="balanced")
lr.fit(xtrain, ytrain)
predictions = lr.predict_proba(xtest)
from sklearn.metrics import roc_auc_score
roc_auc_score(ytest, predictions[:,1])

我得到的分数是0.36，非常差。

有人能帮我吗-

1. 指出我在概念或代码上遗漏了什么
2. 请帮我提高分数

我事先真的很感谢你的帮助。

Answer 1

您的目标是预测两个未连接的节点之间是否存在链接。

首先，提取之间没有链接的节点对。

下一步是从给定的图中隐藏一些边。这是准备训练数据集所必需的。随着社交网络的发展，引入了新的边缘。机器学习模型需要知道图形的演变过程。具有隐藏边的图是在时间t处的图G，并且我们当前的数据集是在时间t+n处的图G。

删除链接或边时，应避免删除任何可能产生未连接节点或网络的边。下一步是为所有未连接的节点对(包括隐藏的节点对(创建特征。

移除的边将被标记为1(正样本(，未连接的节点对将被标记为由0(负样本(。

标记后，使用node2vec算法从图形中提取节点特征。为了计算边的特征，可以将该对节点的特征相加。这些特征将使用逻辑回归模型进行训练。

您可以在节点中添加更多度量值作为值，这样模型就可以根据您的需要预测特征。例如，adamic/adar指数、常见邻居等

因为你已经在训练和测试样本中分割了你的图，你必须找到哪些边具有模型预测的概率。

predictions = lr.predict_proba(xtest)
for i in range(len(df)):
try:
index_in_x_train = np.where(xtrain == x[i])[0][1]
predict_proba = lr.predict_proba(xtrain[index_in_x_train].reshape(1, -1))[:, 1]
print(
f'Probability of nodes {df.iloc[i, 0]} and {df.iloc[i, 1]} to form a link is : {float(predict_proba) * 100 : .2f}%')
except:
continue

try-catch是为了确保我们不会得到索引外错误，因为extrin中的一些数组将为空。

分数低可能是由于隐藏边缘的方式造成的。尝试使用不同的参数和测试大小调整逻辑回归。

此外，您还需要一个更大的数据集，以便对模型进行正确的训练。

你也可以尝试不同的机器学习模型，比如随机森林分类器或多层感知器。