gnnex is to explain the complex GCN model, it shows by drawing the most important edges in whole networks.
Absolutely, there are many different basic GCN models, so you can define your own model.
By rewriting graph.py to define your own GCN Model and rewriting train.py to train your own GCN modeland generate the WEIGHT matrix.
At last, draw the important edges according to WEIGHT matrix above.
pip install gnnex
You should define your own model according to your task to be resolved. By rewriting graph.py to define model and rewriting train.py to train your own model.