/FlyKD

FlyKD, a novel Graph Knowledge Distillation on the Fly with Curriculum Learning for link prediction tasks.

Primary LanguagePythonMIT LicenseMIT

Dear visitors, we really appreciate your stars!

Website and Paper

System Profile:

  • Cuda Profile: 11.8
  • GPU Profile: A5000 & 2080ti

How to replicate experiments:

  1. Install Environment

pip install -r requirements.txt

  1. Download Data (PrimeKG-1GB)

wget -O kg.csv https://dataverse.harvard.edu/api/access/datafile/6180620

  1. Run bash script (results will be stored in Results_e2000/)

. multiple_experiments.sh