├── DataHandler.py ├── Main.py ├── Model.py ├── Models ├── cluster_ours.pickle ├── init.py └── our_base.mod ├── Params.py ├── README.md ├── Utils ├── TimeLogger.py ├── Utils.py └── init.py ├── backbone.py ├── data ├── amazon │ ├── init.py │ ├── trnMat.rar │ ├── tstMat.pkl │ └── valMat.pkl ├── gowalla │ ├── init.py │ ├── trnMat.pkl │ └── tstMat.pkl ├── init.py ├── kindle │ ├── init.py │ ├── trnMat.pkl │ └── tstMat.pkl └── retail_rocket │ ├── init.py │ ├── trnMat.pkl │ └── tstMat.pkl ├── layer_new.py ├── layers.py ├── pictures ├── framework.png ├── init.py └── results.png ├── requirements.txt.txt ├── utils.py └── view_learner.py /DataHandler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/DataHandler.py -------------------------------------------------------------------------------- /Main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/Main.py -------------------------------------------------------------------------------- /Model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/Model.py -------------------------------------------------------------------------------- /Models/cluster_ours.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/Models/cluster_ours.pickle -------------------------------------------------------------------------------- /Models/init.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /Models/our_base.mod: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/Models/our_base.mod -------------------------------------------------------------------------------- /Params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/Params.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/README.md -------------------------------------------------------------------------------- /Utils/TimeLogger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/Utils/TimeLogger.py -------------------------------------------------------------------------------- /Utils/Utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/Utils/Utils.py -------------------------------------------------------------------------------- /Utils/init.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/backbone.py -------------------------------------------------------------------------------- /data/amazon/init.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /data/amazon/trnMat.rar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/data/amazon/trnMat.rar -------------------------------------------------------------------------------- /data/amazon/tstMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/data/amazon/tstMat.pkl -------------------------------------------------------------------------------- /data/amazon/valMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/data/amazon/valMat.pkl -------------------------------------------------------------------------------- /data/gowalla/init.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /data/gowalla/trnMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/data/gowalla/trnMat.pkl -------------------------------------------------------------------------------- /data/gowalla/tstMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/data/gowalla/tstMat.pkl -------------------------------------------------------------------------------- /data/init.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /data/kindle/init.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /data/kindle/trnMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/data/kindle/trnMat.pkl -------------------------------------------------------------------------------- /data/kindle/tstMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/data/kindle/tstMat.pkl -------------------------------------------------------------------------------- /data/retail_rocket/init.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /data/retail_rocket/trnMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/data/retail_rocket/trnMat.pkl -------------------------------------------------------------------------------- /data/retail_rocket/tstMat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/data/retail_rocket/tstMat.pkl -------------------------------------------------------------------------------- /layer_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/layer_new.py -------------------------------------------------------------------------------- /layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/layers.py -------------------------------------------------------------------------------- /pictures/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/pictures/framework.png -------------------------------------------------------------------------------- /pictures/init.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /pictures/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/pictures/results.png -------------------------------------------------------------------------------- /requirements.txt.txt: -------------------------------------------------------------------------------- 1 | Pytorch = 1.12.1 2 | python = 3.9.13 -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/utils.py -------------------------------------------------------------------------------- /view_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/GraphAug/HEAD/view_learner.py --------------------------------------------------------------------------------