T-GCN-PyTorch
这是以下论文中T-GCN的PyTorch实现: 。
可以在找到此存储库的稳定版本。
要求
麻木
matplotlib
大熊猫
火炬
火炬闪电
闪电
模型训练
# GCN
python main.py --model_name GCN --max_epochs 3000 --learning_rate 0.001 --weight_decay 0 --batch_size 64 --hidden_dim 100 --settings supervised --gpus 1
# GRU
python main.py --model_name GRU --max_epochs 3000 --learning_rate 0.001 --weight_decay 1.5e-3 --batch_size 64 --hidden_dim 100 --settings s
1