Paper

Hypergraph-Enhanced cascades prediction via Text-Informed Learning

Requirements

pip install -r requirements.txt

Setting

在 run.py 中设置， 数据集放在了data文件夹下

parser.add_argument('-dataset_name', default='meme')
parser.add_argument('-epoch', default=50)
parser.add_argument('-batch_size', default=64)
parser.add_argument('-emb_dim', default=64)
parser.add_argument('-train_rate', default=0.8)
parser.add_argument('-valid_rate', default=0.1)
parser.add_argument('-max_seq_length', default=200)
parser.add_argument('-step_split', default=10)  # 级联超图的个数
parser.add_argument('-lr', default=0.001)  # 学习率
parser.add_argument('-cl', default=0.1)  # 对比学习强度
parser.add_argument('-doc', type=bool, default=True, help='use text') 
parser.add_argument('-early_stop_step', default=10)  #
parser.add_argument('-preprocess', type=bool, default=True, help="preprocess dataset")

Training

python run.py