• Minjie Wang's avatar
    [NN] Add HeteroGraphConv module for cleaner module definition (#1385) · 3efb5d8e
    Minjie Wang authored
    * Add HeteroGraphConv
    
    * add custom aggregator; some docstring
    
    * debugging
    
    * rm print
    
    * fix some acc bugs
    
    * fix initialization problem in weight basis
    
    * passed tests
    
    * lint
    
    * fix graphconv flag; add error message
    
    * add mxnet heteroconv
    
    * more fix for mx
    
    * lint
    
    * fix torch cuda test
    
    * fix mx test_nn
    
    * add exhaust test for graphconv
    
    * add tf heteroconv
    
    * fix comment
    3efb5d8e
test_nn.py 15.2 KB