├── Constants.py ├── DataConstruct.py ├── DyHGCN.py ├── HGCN.py ├── Metrics.py ├── Optim.py ├── README.md ├── TransformerBlock.py ├── checkpoint └── README.txt ├── data ├── douban │ ├── cascade.txt │ ├── cascadetest.txt │ ├── cascadevalid.txt │ ├── diffusion_graph.csv │ ├── edges.txt │ ├── idx2u.pickle │ └── u2idx.pickle ├── memetracker │ ├── cascade.txt │ ├── cascadetest.txt │ ├── cascadevalid.txt │ ├── diffusion_graph.csv │ ├── idx2u.pickle │ └── u2idx.pickle └── twitter │ ├── cascade.txt │ ├── cascadetest.txt │ ├── cascadevalid.txt │ ├── dev_data.pkl │ ├── diffusion_graph.csv │ ├── edges.txt │ ├── idx2u.pickle │ ├── test_data.pkl │ ├── train_data.pkl │ ├── u2idx.pickle │ └── word_to_ix.pkl └── run.py /Constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/Constants.py -------------------------------------------------------------------------------- /DataConstruct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/DataConstruct.py -------------------------------------------------------------------------------- /DyHGCN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/DyHGCN.py -------------------------------------------------------------------------------- /HGCN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/HGCN.py -------------------------------------------------------------------------------- /Metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/Metrics.py -------------------------------------------------------------------------------- /Optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/Optim.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/README.md -------------------------------------------------------------------------------- /TransformerBlock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/TransformerBlock.py -------------------------------------------------------------------------------- /checkpoint/README.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/douban/cascade.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/douban/cascade.txt -------------------------------------------------------------------------------- /data/douban/cascadetest.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/douban/cascadetest.txt -------------------------------------------------------------------------------- /data/douban/cascadevalid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/douban/cascadevalid.txt -------------------------------------------------------------------------------- /data/douban/diffusion_graph.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/douban/diffusion_graph.csv -------------------------------------------------------------------------------- /data/douban/edges.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/douban/edges.txt -------------------------------------------------------------------------------- /data/douban/idx2u.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/douban/idx2u.pickle -------------------------------------------------------------------------------- /data/douban/u2idx.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/douban/u2idx.pickle -------------------------------------------------------------------------------- /data/memetracker/cascade.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/memetracker/cascade.txt -------------------------------------------------------------------------------- /data/memetracker/cascadetest.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/memetracker/cascadetest.txt -------------------------------------------------------------------------------- /data/memetracker/cascadevalid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/memetracker/cascadevalid.txt -------------------------------------------------------------------------------- /data/memetracker/diffusion_graph.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/memetracker/diffusion_graph.csv -------------------------------------------------------------------------------- /data/memetracker/idx2u.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/memetracker/idx2u.pickle -------------------------------------------------------------------------------- /data/memetracker/u2idx.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/memetracker/u2idx.pickle -------------------------------------------------------------------------------- /data/twitter/cascade.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/twitter/cascade.txt -------------------------------------------------------------------------------- /data/twitter/cascadetest.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/twitter/cascadetest.txt -------------------------------------------------------------------------------- /data/twitter/cascadevalid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/twitter/cascadevalid.txt -------------------------------------------------------------------------------- /data/twitter/dev_data.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/twitter/dev_data.pkl -------------------------------------------------------------------------------- /data/twitter/diffusion_graph.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/twitter/diffusion_graph.csv -------------------------------------------------------------------------------- /data/twitter/edges.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/twitter/edges.txt -------------------------------------------------------------------------------- /data/twitter/idx2u.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/twitter/idx2u.pickle -------------------------------------------------------------------------------- /data/twitter/test_data.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/twitter/test_data.pkl -------------------------------------------------------------------------------- /data/twitter/train_data.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/twitter/train_data.pkl -------------------------------------------------------------------------------- /data/twitter/u2idx.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/twitter/u2idx.pickle -------------------------------------------------------------------------------- /data/twitter/word_to_ix.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/data/twitter/word_to_ix.pkl -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chunyuanY/DyHGCN/HEAD/run.py --------------------------------------------------------------------------------