mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-06 20:04:24 +08:00
4b1fb6816a
* [Model] Heterogeneous graph support for GNNExplainer (#1) * add HeteroGNNExplainer * GNNExplainer for heterogeenous graph * fix typo * variable name cleanup * added HeteroGNNExplainer test * added doc indexing for HeteroGNNExplainer * Update python/dgl/nn/pytorch/explain/gnnexplainer.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update python/dgl/nn/pytorch/explain/gnnexplainer.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update python/dgl/nn/pytorch/explain/gnnexplainer.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update python/dgl/nn/pytorch/explain/gnnexplainer.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update python/dgl/nn/pytorch/explain/gnnexplainer.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update python/dgl/nn/pytorch/explain/gnnexplainer.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update python/dgl/nn/pytorch/explain/gnnexplainer.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update python/dgl/nn/pytorch/explain/gnnexplainer.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update python/dgl/nn/pytorch/explain/gnnexplainer.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update python/dgl/nn/pytorch/explain/gnnexplainer.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update python/dgl/nn/pytorch/explain/gnnexplainer.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update python/dgl/nn/pytorch/explain/gnnexplainer.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update python/dgl/nn/pytorch/explain/gnnexplainer.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update gnnexplainer.py Change DGLHeteroGraph to DGLGraph, and specified parameter inputs * Added ntype parameter to the explainer_node call * responding to @mufeili's comment regarding restoring empty lines at appriopiate places to be consistent with existing practices * responding to @mufeili's comment regarding restoring empty lines at appriopiate places that were missed in the last commit * docstring comments added based on @mufeili suggestions * indorporated @mufeili requested changes related to docstring model declaration. * example model and test_nn.py added for explain_graphs * explain_nodes fixed and fixed the way hetero num nodes and edges are handled * white spaces removed * lint issues fixed * explain_graph model updated * explain nodes model updated * minor fixes related to gpu compatability * cuda support added * simplify WIP * _init_masks for ennexplainer updated to match heterographs * Update * model simplified and docstring comments updated * nits: docstring udpated * lint check issues updated * lint check updated * soem formatting updated * disabling int32 testing for GNNExplainer * Update Co-authored-by: Kangkook Jee <kangkook.jee@gmail.com> Co-authored-by: ahadjawaid <94938815+ahadjawaid@users.noreply.github.com> Co-authored-by: Mufei Li <mufeili1996@gmail.com> Co-authored-by: kxm180046 <kxm180046@utdallas.edu> Co-authored-by: Kunal Mukherjee <kunmukh@gmail.com> Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-26.ap-northeast-1.compute.internal> Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>