Custom Datasets + Module Loading Error

zfjsail / gae-pytorch

Graph Auto-Encoder in PyTorch

MIT License

417 stars 79 forks source link

Custom Datasets + Module Loading Error #2

Closed jlevy44 closed 5 years ago

jlevy44 commented 5 years ago

Can I use this framework on my own custom datasets?

Also, I'm experiencing an issue, when I try to run: python gae/train.py

I get:

Traceback (most recent call last): File "gae/train.py", line 12, in from gae.model import GCNModelVAE ModuleNotFoundError: No module named 'gae'

Yet this command works interactively.

jlevy44 commented 5 years ago

Path works fine for me right now. Forgot to export the python path. Still need help with the custom dataset. Limited setup instructions.

jlevy44 commented 5 years ago

I have a 30x30 adjacency matrix with some covariates. I'm looking to rebuild this adjacency matrix while being able to look at the latent space of the graph. Is that possible with this framework? I've been able to get node-level embeddings here, but nothing yet on the graph level.