AUT-MSAM

PyTorch implementation of Hot Topic-Aware Retweet Prediction with Masked Self-attentive Model

Requirements

• Python 3.6
• Pytorch 0.4.0
• In addition, anyone who want to run the model should download the dataset and trained model parameters from https://pan.baidu.com/s/14w2P1RxE9zxHKpjKL5CY9Q
• You can also download the dataset from ^_^ https://drive.google.com/drive/folders/1g1Ns866pSoY6v4Hd_TkFXxwIcfQGuy4b?usp=sharing