Hi Docherty! I tried your code and encountered the error on class GraphN2V's definition:

I'm using node2vec v0.4.2, networkx v2.5, and other packages seem to have nothing to do with this problem. Looking forward for your reply😁
B.T.W. The rest of the code is really performant!