Model storage

[ana-kuznetsova]

Hello Thomas! Using the library I did not figure out how to store the model for subsequent use and how to get embeddings from it. Could you help me out with this?

Thank you!

[tkipf]

For model saving/loading, tf.train.Saver should work: https://www.tensorflow.org/guide/saved_model

You can get embeddings from a trained model via model.z_mean or model.embeddings (the latter one is a sample from a Gaussian latent variable for the VGAE model).

[ana-kuznetsova]

Thank you!

[ioannisa92]

@ana-kuznetsova were you able to save/use the model on your own graphs?

Any help/sample code would be very useful.

Thanks