Skip to content

Junshan-Wang/ContinualGNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 

Repository files navigation

Streaming Graph Neural Networks via Continual Learning

Code for Streaming Graph Neural Networks via Continual Learning(CIKM 2020). ContinualGNN is a streaming graph neural network based on continual learning so that the model is trained incrementally and up-to-date node representations can be obtained at each time step.

Requirements

  • python = 3.8.5
  • pytorch = 1.7.1
  • scikit-learn = 0.23.2

Usages

  • ContinualGNN (proposed model) on Cora:
cd src/
python main_stream.py --data=cora --new_ratio=0.8 --memory_size=250 --ewc_lambda=80.0 
  • OnlineGNN (lower bound) on Cora:
python main_stream.py --data=cora

If using cuda, set --cuda.

About

Streaming Graph Neural Networks via Continual Learning (CIKM 2020)

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published