探索图注意力网络架构和图卷积架构以对蛋白质-蛋白质相互作用数据集中的节点进行分类。 在pytorch中实现。
如何运行:
安装requirements.txt中所述的requirements.txt
要运行火车脚本,请使用以下命令:
python train.py --model_type=<'GAT' for Graph Attention Network, 'GCN' for Graph Convolution network architecture>
--input_dir=
--output_dir=<dir to save the model, e.g. 'saved_mo
2021-09-27 17:24:00
8.49MB
Python
1