Link Prediction Python Packages

pykeen

🤖 A Python library for learning and evaluating knowledge graph embeddings

stellargraph

StellarGraph - Machine Learning on Graphs

cogdl

CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)

heterosplit

Leakage-safe entity-disjoint splits and audits for heterogeneous link-prediction datasets (PyTorch Geometric-compatible).

deltakg

PromptKG Family: a Gallery of Prompt Learning & KG-related research works, toolkits, and paper-list.

linkpred

Easy link prediction tool

gnn-augment

pip install gnn-augment — graph-augmented GNNs (GCN, GraphSAGE, GAT, GIN) for semi-supervised node classification: replace the adjacency with Katz / Rooted PageRank / Graph Gravity / LLM text-embedding views, multi-view training, sparse mode for large graphs (arXiv:1905.09087)

neuralkg

[Tool] For Knowledge Graph Representation Learning

grakn-kglib

TypeDB-ML is the Machine Learning integrations library for TypeDB

bigraph

Bipartite-network link prediction in Python

evalne

Source code for EvalNE, a Python library for evaluating Network Embedding methods.

edgelist-mapper

📊Maps nodes and edges of a multi-relational graph to integer

complex-network-link-prediction

A python library for link prediction in social networks

graph-dependence

Reference implementation for separating graph dependence from graph advantage in graph-ablation studies.

ffscraper

Yet another set of scraping tools for FanFiction.Net

dicn

An implementation of the Direct-Indirect Common Neighbors algorithm

relationalstats

A practical Python toolkit for applied and statistical social network analysis, with research-grade documentation and transparent methodological boundaries.

typedb-ml

TypeDB-ML is the Machine Learning integrations library for TypeDB

kgeditor

PromptKG Family: a Gallery of Prompt Learning & KG-related research works, toolkits, and paper-list.

gnn4lp

gnn for link prediction，图神经网络用于链接预测。

gcn4lp

gcn for prediction of protein interactions，图卷积用于蛋白质相互作用。

stellargraph-mvisani

StellarGraph - Machine Learning on Graphs