Hi, thank you for sharing this nice work. My question is, does your repo have a script to generate the hetero-graph visualization given in Figure 1 of the paper? Thanks