在 Python 中进行链路预测可以使用一些图论库, 例如 NetworkX。
在使用 NetworkX 之前, 需要先安装它:
pipinstall networkx
然后可以使用下面的代码来构建一个简单的链路预测模型:
import networkx as nx
# 创建一个空的无向图
G = nx.Graph()
# 添加边
G.add_edge('A', 'B')
G.add_edge('B', 'C')
G.add_edge('C', 'D')
# 预测 'A' 和 'D' 之间是否存在链路
print(nx.has_path(G, 'A', 'D'))