画不同的颜色来显示边缘的重要性
gnnex的Python项目详细描述
简介
gnnex是为了解释复杂的GCN模型, 它通过绘制整个网络中最重要的边来表示。在
当然,有许多不同的基本GCN模型,所以您可以定义自己的模型 模型。在
重写图形.py定义自己的GCN模型并重写火车.py训练你的 拥有GCN模型并生成权重矩阵。在
最后,根据上述权重矩阵,画出重要的边缘。在
已安装
keras==2.3.1
numpy==1.17.2
pandas==0.25.3
matplotlib==3.1.1
scipy==1.3.2
networkx==2.2
seaborn==0.9.0
使用
pip安装gnnex
基本模型
您应该根据要解决的任务定义自己的模型。重写图形.py定义模型和重写火车.py训练你自己的模特。在
- 项目
标签: