├── DataLoader.py ├── Irvine.chkpt ├── Lastfm.chkpt ├── Meme.chkpt ├── README.md ├── data ├── irvine │ ├── cascade.txt │ ├── cascadetest.txt │ ├── idx2u.pickle │ └── u2idx.pickle ├── lastfm │ ├── cascade.txt │ ├── cascadetest.txt │ ├── idx2u.pickle │ └── u2idx.pickle └── memetracker │ ├── cascade.txt │ ├── cascadetest.txt │ ├── idx2u.pickle │ └── u2idx.pickle ├── evaluate.py ├── train.py └── transformer ├── Constants.py ├── Layers.py ├── Models.py ├── Modules.py ├── Optim.py ├── SubLayers.py ├── Translator.py └── __init__.py /DataLoader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/DataLoader.py -------------------------------------------------------------------------------- /Irvine.chkpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/Irvine.chkpt -------------------------------------------------------------------------------- /Lastfm.chkpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/Lastfm.chkpt -------------------------------------------------------------------------------- /Meme.chkpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/Meme.chkpt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/README.md -------------------------------------------------------------------------------- /data/irvine/cascade.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/data/irvine/cascade.txt -------------------------------------------------------------------------------- /data/irvine/cascadetest.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/data/irvine/cascadetest.txt -------------------------------------------------------------------------------- /data/irvine/idx2u.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/data/irvine/idx2u.pickle -------------------------------------------------------------------------------- /data/irvine/u2idx.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/data/irvine/u2idx.pickle -------------------------------------------------------------------------------- /data/lastfm/cascade.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/data/lastfm/cascade.txt -------------------------------------------------------------------------------- /data/lastfm/cascadetest.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/data/lastfm/cascadetest.txt -------------------------------------------------------------------------------- /data/lastfm/idx2u.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/data/lastfm/idx2u.pickle -------------------------------------------------------------------------------- /data/lastfm/u2idx.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/data/lastfm/u2idx.pickle -------------------------------------------------------------------------------- /data/memetracker/cascade.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/data/memetracker/cascade.txt -------------------------------------------------------------------------------- /data/memetracker/cascadetest.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/data/memetracker/cascadetest.txt -------------------------------------------------------------------------------- /data/memetracker/idx2u.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/data/memetracker/idx2u.pickle -------------------------------------------------------------------------------- /data/memetracker/u2idx.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/data/memetracker/u2idx.pickle -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/evaluate.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/train.py -------------------------------------------------------------------------------- /transformer/Constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/transformer/Constants.py -------------------------------------------------------------------------------- /transformer/Layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/transformer/Layers.py -------------------------------------------------------------------------------- /transformer/Models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/transformer/Models.py -------------------------------------------------------------------------------- /transformer/Modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/transformer/Modules.py -------------------------------------------------------------------------------- /transformer/Optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/transformer/Optim.py -------------------------------------------------------------------------------- /transformer/SubLayers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/transformer/SubLayers.py -------------------------------------------------------------------------------- /transformer/Translator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/transformer/Translator.py -------------------------------------------------------------------------------- /transformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangchengbupt/NeuralDiffusionModel/HEAD/transformer/__init__.py --------------------------------------------------------------------------------