/RGExplainer_new

Editing codebase for the paper Reinforcement Learning Enhanced Explainer for Graph Neural Networks

Primary LanguagePython

This is a customized Pytorch implementation of RG-Explainer for the paper Explaining the Explainers in Graph Neural Networks: a Comparative Study.

You can find the original code here

Tip to run the code

python run_RG_explainer.py --paper SURVEY --dataset GRID --model CHEB --seed 42 --n_epochs 50 --size_reg 0.01 --sim_reg 1.0 --radius_penalty 0.1 > Survey_Grid_Cheb.txt 2>&1