/CasGT

This is the final project of DATA130007.01 (Social Network Mining).

Primary LanguagePython

CasGT

This is the final project of DATA130007.01 (Social Network Mining).

Twitter数据

data link

运行方式（以CasGT_10为例）

  1. 进入GRU文件夹，将Twitter数据中的data_small数据加入文件夹中，运行

python train.py --epoch 10

  1. 将保存的CasGT_10模型model.pth与data_small数据加入TopoLSTM文件夹中，运行

python train.py --data_dir 'data_small'

  1. 模型测试

python test.py --data_dir 'data_small'

  1. 结果评估

python evaluation.py