
Graph-Transformer:基于Pytorch和Tensorflow的图形分类变压器
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
Graph-Transformer是一款结合了Pytorch与TensorFlow框架的先进图形分类工具。它运用了Transformer模型对图数据进行高效处理,适用于深度学习领域的复杂网络分析任务。
图分类用变压器
这一计划规定,在描述了我们U2GNN执行后,我们充分利用了变压器自重视网络构建一个先进的聚合函数以学习图形表示。
使用要求:
- 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
```
全部评论 (0)
还没有任何评论哟~


