文件名称:GNNs用于蛋白质-蛋白质相互作用
文件大小:8.49MB
文件格式:ZIP
更新时间:2024-03-01 15:28:40
Python
探索图注意力网络架构和图卷积架构以对蛋白质-蛋白质相互作用数据集中的节点进行分类。 在pytorch中实现。
如何运行:
安装requirements.txt中所述的requirements.txt
要运行火车脚本,请使用以下命令:
python train.py --model_type=<'GAT' for Graph Attention Network, 'GCN' for Graph Convolution network architecture>
--input_dir=
【文件预览】:
GNNs-for-protein-protein-interaction-master
----core()
--------models.py(2KB)
----requirements.txt(132B)
----data()
--------dataloader.py(2KB)
--------ppi_dataset()
----utils()
--------preprocessing.py(3KB)
----README.md(990B)
----evaluate.py(3KB)
----saved_models()
--------best_model_GAT.pt(528KB)
--------best_model_GCN.pt(175KB)
----train.py(6KB)