├── .gitignore ├── LICENSE ├── README.md ├── args.py ├── baselines ├── __init__.py ├── graphens.py ├── graphsmote.py ├── renode.py └── reweight.py ├── bat.py ├── config.yaml ├── data_utils.py ├── dev ├── args.py ├── data_utils.py ├── dataloader.py ├── ens_nets │ ├── __init__.py │ ├── gat.py │ ├── gcn.py │ └── sage.py ├── experiment_example.ipynb ├── losses │ ├── __init__.py │ ├── balanced_softmax.py │ └── ce.py ├── models │ ├── __init__.py │ ├── gens.py │ ├── graphsmote.py │ ├── pc_softmax.py │ ├── renode.py │ ├── reweight.py │ └── tam.py ├── nets │ ├── __init__.py │ ├── gat.py │ ├── gcn.py │ └── sage.py ├── results │ └── IR(10)-rep(2)-epoch(100)-gnn['GCN']-data['Cora']-bsl['vanilla', 'reweight', 'resample', 'renode', 'smote', 'graphsmote', 'graphens'].csv ├── routing.py └── utils.py ├── nets ├── __init__.py ├── gat.py ├── gcn.py ├── gens_networks │ ├── __init__.py │ ├── gat.py │ ├── gcn.py │ └── sage.py └── sage.py ├── train.py ├── train_example.ipynb ├── trainer.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/README.md -------------------------------------------------------------------------------- /args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/args.py -------------------------------------------------------------------------------- /baselines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/baselines/__init__.py -------------------------------------------------------------------------------- /baselines/graphens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/baselines/graphens.py -------------------------------------------------------------------------------- /baselines/graphsmote.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/baselines/graphsmote.py -------------------------------------------------------------------------------- /baselines/renode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/baselines/renode.py -------------------------------------------------------------------------------- /baselines/reweight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/baselines/reweight.py -------------------------------------------------------------------------------- /bat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/bat.py -------------------------------------------------------------------------------- /config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/config.yaml -------------------------------------------------------------------------------- /data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/data_utils.py -------------------------------------------------------------------------------- /dev/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/args.py -------------------------------------------------------------------------------- /dev/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/data_utils.py -------------------------------------------------------------------------------- /dev/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/dataloader.py -------------------------------------------------------------------------------- /dev/ens_nets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/ens_nets/__init__.py -------------------------------------------------------------------------------- /dev/ens_nets/gat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/ens_nets/gat.py -------------------------------------------------------------------------------- /dev/ens_nets/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/ens_nets/gcn.py -------------------------------------------------------------------------------- /dev/ens_nets/sage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/ens_nets/sage.py -------------------------------------------------------------------------------- /dev/experiment_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/experiment_example.ipynb -------------------------------------------------------------------------------- /dev/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/losses/__init__.py -------------------------------------------------------------------------------- /dev/losses/balanced_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/losses/balanced_softmax.py -------------------------------------------------------------------------------- /dev/losses/ce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/losses/ce.py -------------------------------------------------------------------------------- /dev/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/models/__init__.py -------------------------------------------------------------------------------- /dev/models/gens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/models/gens.py -------------------------------------------------------------------------------- /dev/models/graphsmote.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/models/graphsmote.py -------------------------------------------------------------------------------- /dev/models/pc_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/models/pc_softmax.py -------------------------------------------------------------------------------- /dev/models/renode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/models/renode.py -------------------------------------------------------------------------------- /dev/models/reweight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/models/reweight.py -------------------------------------------------------------------------------- /dev/models/tam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/models/tam.py -------------------------------------------------------------------------------- /dev/nets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/nets/__init__.py -------------------------------------------------------------------------------- /dev/nets/gat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/nets/gat.py -------------------------------------------------------------------------------- /dev/nets/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/nets/gcn.py -------------------------------------------------------------------------------- /dev/nets/sage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/nets/sage.py -------------------------------------------------------------------------------- /dev/results/IR(10)-rep(2)-epoch(100)-gnn['GCN']-data['Cora']-bsl['vanilla', 'reweight', 'resample', 'renode', 'smote', 'graphsmote', 'graphens'].csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/results/IR(10)-rep(2)-epoch(100)-gnn['GCN']-data['Cora']-bsl['vanilla', 'reweight', 'resample', 'renode', 'smote', 'graphsmote', 'graphens'].csv -------------------------------------------------------------------------------- /dev/routing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/routing.py -------------------------------------------------------------------------------- /dev/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/dev/utils.py -------------------------------------------------------------------------------- /nets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/nets/__init__.py -------------------------------------------------------------------------------- /nets/gat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/nets/gat.py -------------------------------------------------------------------------------- /nets/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/nets/gcn.py -------------------------------------------------------------------------------- /nets/gens_networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/nets/gens_networks/__init__.py -------------------------------------------------------------------------------- /nets/gens_networks/gat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/nets/gens_networks/gat.py -------------------------------------------------------------------------------- /nets/gens_networks/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/nets/gens_networks/gcn.py -------------------------------------------------------------------------------- /nets/gens_networks/sage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/nets/gens_networks/sage.py -------------------------------------------------------------------------------- /nets/sage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/nets/sage.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/train.py -------------------------------------------------------------------------------- /train_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/train_example.ipynb -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/trainer.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhiningLiu1998/BAT/HEAD/utils.py --------------------------------------------------------------------------------