spektral icon indicating copy to clipboard operation
spektral copied to clipboard

[Feature Request] Could you add Unspervised GraphSage and Embedding output

Open zhenghang opened this issue 3 years ago • 1 comments

Tks about your code. Could you add Unspervised GraphSage and embedding output when model is trained

zhenghang avatar Mar 08 '21 09:03 zhenghang

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

danielegrattarola avatar Mar 08 '21 18:03 danielegrattarola