Skip to content

Latest commit

 

History

12 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 

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

41 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages