saving ''Embedded Mean Field'' to a txt file?

[donnate]

Hello,

Thank you very much for your very interesting paper and code!
I have been trying to apply your method and in particular, I was hoping to study the output of the Embedded Mean Field algorithm (algorithm 1) of your paper. I admit to being a little lost with all of the classes and types: I was wondering if there was any way of saving each of the node embeddings to a txt file (before they are collapsed for graph embedding)?
Thank you very much for your time and help!
Best regards,

Claire

[Hanjun-Dai]

Hi there,

1. If you prefer this C++ code, you can check out the example/mnist, in which I dump the prediction to file. You should be able to do it similarly for embeddings.

2. If you prefer python, I've created a pytorch implementation:
   https://github.com/Hanjun-Dai/pytorch_structure2vec
   where getting embeddings should be straightforward.