This package contains code for "Bayesian Neighborhood Adaptation for Graph Neural Networks"

Required packages:
torch
numpy
dgl


To run:

python run.py --fs True --dataset cora --truncation 8 --kld_weight 1.0
python run.py --fs True --dataset citeseer --truncation 8 --kld_weight 2.0
python run.py --dataset pubmed --truncation 8 --kld_weight 1.0



