thunlp / OpenNE

An Open-Source Package for Network Embedding (NE)
MIT License
1.68k stars 485 forks source link

The outputs of GCN are not embedding #110

Closed imJiawen closed 3 years ago

imJiawen commented 3 years ago

Hi there, Thanks for your hard work. I want to use the embedding generated by GCN, but its output embedding size is not what I set by "--dim" option. After checking I found that the dimension of output embedding produced by GCN is the same as the number of labels.

It seems that I cannot set the output embedding size of GCN as I want. Is it possible to solve this problem? Thank you in advance.

Qian-Hao commented 3 years ago

I have the same problem! Have you solved the problem?

imJiawen commented 3 years ago

I have the same problem! Have you solved the problem?

Hi @Qian-Hao, I am afraid that the trained GCN embedding cannot be used directly for other downstream tasks. Instead, it can only perform node classification task and returns the accuracy of it. I have used other graph embedding methods as an alternative. :)