pyhton 3.10.11
torch 2.1.0
torch-geometric 2.3.1
可以下载原始数据集进行本地处理,也可以直接下载预处理后的数据集,置于datasets/new
路径下。
可以下载完整conceptnet进行本地处理,也可以联系作者[email protected]获取本地处理后的知识库,置于datasets/inter-file
路径下。
下载预训练的中文词嵌入模型,置于model/pre_train
路径下。
python run.py --slide 3 --in_channels 300 --hidden_channels 768 --num_heads 3 --layers 3 --dropout 0.4 --lr 6e-6 --weight_decay 5e-5 --batch_size 16