This is the PyTorch implementation for our Graph-oriented Influence Function.
The code runs well under python 3.6.10. The required packages are as follows:
For regular unlearning task (e.g., Edge Unlearning with ratio 0.05 on Cora using GCN)
python main.py --dataset_name cora --target_model GCN --is_train_target_model True --exp Unlearn --method GIF --is_use_node_feature True --num_runs 10 --unlearn_task edge --unlearn_ratio 0.05 --iteration 100 --scale 500
For adversarial attack experiments (e.g., ratio=0.5 on Cora using GCN)
python main.py --dataset_name cora --target_model GCN --is_train_target_model True --exp Attack --method GIF --is_use_node_feature True --num_runs 10 --unlearn_task edge --unlearn_ratio 0.5 --iteration 100 --scale 500