BiGCN踩坑记录
论文名:Rumor Detection on Social Media with Bi-Directional Graph Convolutional Networks
作者:Tian Bian,Xi Xiao,Tingyang Xu,Peilin Zhao,Wenbing Huang,Yu Rong,Junzhou Huang
代码地址:https://github.com/TianBian95/BiGCN
1.创建环境
不说废话,首先是环境创建的要注意的事情,python版本3.7就能用,torch要安装跟python版本对应的1.4版本(去pytorch官网找早期版本),requirement.txt文件里面的环境就去https://pytorch-geometric.com/whl/下载对应于你python版本和系统环境的whl文件,我选择的如下所示:
torch_geometric 就直接使用
pip install torch_geometric==1.3.2
安装,tqdm和joblib也是按照他给的版本号安装
pip install tqdm==4.40.0
pip install joblib==0.14.1
至此,环境安装结束。
- 数据获取
选中的三个文件夹需要手动创建,不然在运行生成图数据时会直接G。
生成数据代码就是main.sh中的三行代码(如图所示):
(微博那个数据巨慢,可以先搞推特数据先跑…,再遇到坑继续记录…) - 训练
我在训练twitter15时报如下错误:
我估计是python版本的问题,解决办法参考:
https://blog.csdn.net/m0_37052320/article/details/118368656