• Da Zheng's avatar
    [KG] PBG's way of constructing negative edges (#1159) · 1022d5d6
    Da Zheng authored
    * attach positive.
    
    * add neg_deg_sample.
    
    * add comment.
    
    * add neg_deg_sample for eval.
    
    * change the edge sampler.
    
    * rename edge sampler in KG.
    
    * allow specifying chunk size and negative sample size separately.
    
    * fix bugs in KG.
    
    * add check in sampler.
    
    * add more checks.
    
    * fix
    
    * add comment.
    
    * add comments.
    1022d5d6
train.py 19 KB