├── .gitignore ├── DataHandler.py ├── Datasets ├── sparse_amazon │ ├── trnMat.pkl │ ├── tstMat.pkl │ └── valMat.pkl ├── sparse_gowalla │ ├── trnMat.pkl │ ├── tstMat.pkl │ └── valMat.pkl └── sparse_yelp │ ├── trnMat.pkl │ ├── tstMat.pkl │ └── valMat.pkl ├── Main.py ├── Model.py ├── Params.py ├── README.md ├── Supplementary Material.pdf ├── Utils ├── TimeLogger.py ├── Utils.py └── __pycache__ │ ├── TimeLogger.cpython-39.pyc │ └── Utils.cpython-39.pyc ├── __pycache__ ├── DataHandler.cpython-39.pyc ├── Model.cpython-39.pyc ├── Params.cpython-39.pyc ├── main_gcn_imp.cpython-39.pyc ├── pruning.cpython-37.pyc ├── pruning.cpython-38.pyc ├── pruning.cpython-39.pyc ├── pruning_gcn.cpython-38.pyc └── pruning_gcn.cpython-39.pyc ├── develop-environment.md ├── example_scripts ├── example_train_final_student_lightgnn_amazon.sh ├── example_train_final_student_lightgnn_gowalla.sh ├── example_train_final_student_lightgnn_yelp.sh ├── example_train_intermediate_model_lightgnn_amazon.sh ├── example_train_intermediate_model_lightgnn_gowalla.sh ├── example_train_intermediate_model_lightgnn_yelp.sh ├── example_train_teacher_model_amazon.sh ├── example_train_teacher_model_gowalla.sh └── example_train_teacher_model_yelp.sh ├── inModels └── download_demo_ckpts.sh ├── logs └── .gitkeep ├── outModels ├── amazon │ ├── example1 │ │ ├── checkpoints │ │ │ └── .gitkeep │ │ └── history │ │ │ └── .gitkeep │ ├── example2 │ │ ├── checkpoints │ │ │ └── .gitkeep │ │ └── history │ │ │ └── .gitkeep │ └── example3 │ │ ├── checkpoints │ │ └── .gitkeep │ │ └── history │ │ └── .gitkeep ├── gowalla │ ├── example1 │ │ ├── checkpoints │ │ │ └── .gitkeep │ │ └── history │ │ │ └── .gitkeep │ ├── example2 │ │ ├── checkpoints │ │ │ └── .gitkeep │ │ └── history │ │ │ └── .gitkeep │ └── example3 │ │ ├── checkpoints │ │ └── .gitkeep │ │ └── history │ │ └── .gitkeep └── yelp │ ├── example1 │ ├── checkpoints │ │ └── .gitkeep │ └── history │ │ └── .gitkeep │ ├── example2 │ ├── checkpoints │ │ └── .gitkeep │ └── history │ │ └── .gitkeep │ └── example3 │ ├── checkpoints │ └── .gitkeep │ └── history │ └── .gitkeep ├── pretrainTeacher.py └── pruning_gcn.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.mod 2 | !.gitkeep 3 | -------------------------------------------------------------------------------- /DataHandler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/DataHandler.py -------------------------------------------------------------------------------- /Datasets/sparse_amazon/trnMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Datasets/sparse_amazon/trnMat.pkl -------------------------------------------------------------------------------- /Datasets/sparse_amazon/tstMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Datasets/sparse_amazon/tstMat.pkl -------------------------------------------------------------------------------- /Datasets/sparse_amazon/valMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Datasets/sparse_amazon/valMat.pkl -------------------------------------------------------------------------------- /Datasets/sparse_gowalla/trnMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Datasets/sparse_gowalla/trnMat.pkl -------------------------------------------------------------------------------- /Datasets/sparse_gowalla/tstMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Datasets/sparse_gowalla/tstMat.pkl -------------------------------------------------------------------------------- /Datasets/sparse_gowalla/valMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Datasets/sparse_gowalla/valMat.pkl -------------------------------------------------------------------------------- /Datasets/sparse_yelp/trnMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Datasets/sparse_yelp/trnMat.pkl -------------------------------------------------------------------------------- /Datasets/sparse_yelp/tstMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Datasets/sparse_yelp/tstMat.pkl -------------------------------------------------------------------------------- /Datasets/sparse_yelp/valMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Datasets/sparse_yelp/valMat.pkl -------------------------------------------------------------------------------- /Main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Main.py -------------------------------------------------------------------------------- /Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Model.py -------------------------------------------------------------------------------- /Params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Params.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/README.md -------------------------------------------------------------------------------- /Supplementary Material.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Supplementary Material.pdf -------------------------------------------------------------------------------- /Utils/TimeLogger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Utils/TimeLogger.py -------------------------------------------------------------------------------- /Utils/Utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Utils/Utils.py -------------------------------------------------------------------------------- /Utils/__pycache__/TimeLogger.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Utils/__pycache__/TimeLogger.cpython-39.pyc -------------------------------------------------------------------------------- /Utils/__pycache__/Utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/Utils/__pycache__/Utils.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/DataHandler.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/__pycache__/DataHandler.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/Model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/__pycache__/Model.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/Params.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/__pycache__/Params.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/main_gcn_imp.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/__pycache__/main_gcn_imp.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/pruning.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/__pycache__/pruning.cpython-37.pyc -------------------------------------------------------------------------------- /__pycache__/pruning.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/__pycache__/pruning.cpython-38.pyc -------------------------------------------------------------------------------- /__pycache__/pruning.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/__pycache__/pruning.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/pruning_gcn.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/__pycache__/pruning_gcn.cpython-38.pyc -------------------------------------------------------------------------------- /__pycache__/pruning_gcn.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/__pycache__/pruning_gcn.cpython-39.pyc -------------------------------------------------------------------------------- /develop-environment.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/develop-environment.md -------------------------------------------------------------------------------- /example_scripts/example_train_final_student_lightgnn_amazon.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/example_scripts/example_train_final_student_lightgnn_amazon.sh -------------------------------------------------------------------------------- /example_scripts/example_train_final_student_lightgnn_gowalla.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/example_scripts/example_train_final_student_lightgnn_gowalla.sh -------------------------------------------------------------------------------- /example_scripts/example_train_final_student_lightgnn_yelp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/example_scripts/example_train_final_student_lightgnn_yelp.sh -------------------------------------------------------------------------------- /example_scripts/example_train_intermediate_model_lightgnn_amazon.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/example_scripts/example_train_intermediate_model_lightgnn_amazon.sh -------------------------------------------------------------------------------- /example_scripts/example_train_intermediate_model_lightgnn_gowalla.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/example_scripts/example_train_intermediate_model_lightgnn_gowalla.sh -------------------------------------------------------------------------------- /example_scripts/example_train_intermediate_model_lightgnn_yelp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/example_scripts/example_train_intermediate_model_lightgnn_yelp.sh -------------------------------------------------------------------------------- /example_scripts/example_train_teacher_model_amazon.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/example_scripts/example_train_teacher_model_amazon.sh -------------------------------------------------------------------------------- /example_scripts/example_train_teacher_model_gowalla.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/example_scripts/example_train_teacher_model_gowalla.sh -------------------------------------------------------------------------------- /example_scripts/example_train_teacher_model_yelp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/example_scripts/example_train_teacher_model_yelp.sh -------------------------------------------------------------------------------- /inModels/download_demo_ckpts.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/inModels/download_demo_ckpts.sh -------------------------------------------------------------------------------- /logs/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/amazon/example1/checkpoints/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/amazon/example1/history/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/amazon/example2/checkpoints/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/amazon/example2/history/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/amazon/example3/checkpoints/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/amazon/example3/history/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/gowalla/example1/checkpoints/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/gowalla/example1/history/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/gowalla/example2/checkpoints/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/gowalla/example2/history/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/gowalla/example3/checkpoints/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/gowalla/example3/history/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/yelp/example1/checkpoints/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/yelp/example1/history/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/yelp/example2/checkpoints/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/yelp/example2/history/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/yelp/example3/checkpoints/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /outModels/yelp/example3/history/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretrainTeacher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/pretrainTeacher.py -------------------------------------------------------------------------------- /pruning_gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/LightGNN/HEAD/pruning_gcn.py --------------------------------------------------------------------------------