Skip to content

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

License

Notifications You must be signed in to change notification settings

Drug-Repurposing-GNN/FlyKD

Repository files navigation

Dear visitors, we really appreciate your stars!

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

About

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

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •