How can I visualize diffpool process?

[daeunni]

Thank you for your nice work.

Like in your paper fig.2 , How can I visualize diffpooling process?
What method did you use?

Best,
Daeun Lee

[Skyorca]

You can use networkx. Roughly speaking, after assigning each node with an id and using a mapping from id to color, you can draw graphs with node colored. The dash circle may be added manually :)