图分类用变压器 这一计划规定,在描述了我们U2GNN执行,我们充分利用了变压器自重视网络,构建一个先进的聚合函数学习图形表示。 用法 消息 17-05-2020:更新Pytorch(1.5.0)实施。 要求 Python 3.x Tensorflow 1.14 张量2张量1.13 Networkx 2.3 Scikit学习0.21.2 训练 U2GNN$ python train_U2GNN_Sup.py --dataset IMDBBINARY --batch_size 4 --ff_hidden_size 1024 --fold_idx 1 --num_neighbors 8 --num_sampled 512 --num_epochs 50 --num_timesteps 4 --learning_rate 0.0005 --model_name IMDBBINARY_bs
1