Hi I wrote some code with dgl to reproduce the results of this paper. Not sure if there is anyone interested in this.
A sample of results:
I am stuck with oom problem currently. And my laptop could only run some real small samples. The structure used in that paper is very simple though.