search

THUDM / GATNE

Source code and dataset for KDD 2019 paper "Representation Learning for Attributed Multiplex Heterogeneous Network"
MIT License
525 stars 141 forks source link

What method did you used to make link prediction? #40

Closed dsj96 closed 4 years ago

dsj96 commented 4 years ago

When got the embedding of the nodes on every type of edges, I am not clearly with how to used the embedding to make the prediction? I would appreciate it if you could answer this question.

cenyk1230 commented 4 years ago

Hi @dsj96,

We use the cosine similarity of node embeddings to make the prediction. You can see the evaluation part of our code for the detail.