• Mufei Li's avatar
    [NN] Fix GCN module (#99) · 2758c249
    Mufei Li authored
    1. Update `examples/pytorch/gcn` and `python/dgl/nn/pytorch` based on the latest APIs
    2. Add full support for dropout in `examples/pytorch/gcn` and `python/dgl/nn/pytorch`
    3. Rename `GCN` class in `python/dgl/nn/pytorch` to be `GraphConvolutionLayer` class
    4. Make node field an argument that can be configured by users in GraphConvolutionLayer
    
    Note that adjacency normalization has not been supported yet in the examples. 
    2758c249
gcn_spmv.py 4.2 KB