spektral
spektral copied to clipboard
[Feature Request] Could you add Unspervised GraphSage and Embedding output
Tks about your code. Could you add Unspervised GraphSage and embedding output when model is trained
Hi,
can you elaborate on what you mean by "embedding output"?
If you want to retrieve the output of a model after training usually it's enough to call model(inputs)
.
If you are using a Loader, then inputs
can be retrieved as follows:
batch = loader.__next__()
inputs = batch[0]
outputs = model(inputs)
Cheers