• Minjie Wang's avatar
    [Model] RGCN mini-batch training; Better bipartite graph support (#1337) · 6111ea46
    Minjie Wang authored
    * change the model to use node embedding only
    
    * minibatch training
    
    * add readme
    
    * small fix
    
    * backward-compatible
    
    * backward compatible
    
    * modify to_block; rgcn changed
    
    * fix
    
    * fix transform
    
    * fix bug in unittest script
    
    * docstring
    
    * fix lint
    
    * add tests
    
    * address comments; fix offline eval
    
    * gitignore
    6111ea46
task_unit_test.sh 1020 Bytes