By Ariane Marandon.

This repository contains our implementation of the conformal link prediction method introduced in the paper "Conformal link prediction for FDR control"(https://arxiv.org/abs/2306.14693).

Requirements:

numpy==1.24.2
torch==2.0.0
torch_geometric==2.3.0

See notebook in this repository for a tutorial.

Parts of the code are taken from the implementation of the link prediction algorithm "SEAL":

M. Zhang and Y. Chen, Link Prediction Based on Graph Neural Networks, Advances in Neural Information Processing Systems (NIPS-18). arxiv