search

THUDM / GATNE

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

关于 NSLoss 的问题 #108

Open JXT218 opened 3 years ago

JXT218 commented 3 years ago

您好，

我注意到在dgl 的 implementation, negative sample 并不是uniform 的，而是

[ (log(i+2) - log(i+1)) / log(num_nodes + 1)]

当i 越来越大，sample 到的概率越小，可是node 的 index 是随机生成，为什么要让sample的概率和node的顺序有关呢？

见如下链接 https://github.com/dmlc/dgl/blob/31f4483af7eaced74540bfbea373155b16720183/examples/pytorch/GATNE-T/src/main.py#L187

cenyk1230 commented 3 years ago

Hi @JXT218 节点的index是会根据random walks里的出现次数来重编号的，编号越小出现次数越多。