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.
- python = 3.8.5
- pytorch = 1.7.1
- scikit-learn = 0.23.2
- 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
.