Skip to content

GitLab

Unverified Commit 97a6d740 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[doc] fix error in example results (#3735) 


Co-authored-by: default avatarMufei Li <mufeili1996@gmail.com>
parent 4f7ad258
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
python/dgl/nn/pytorch/explain/gnnexplainer.py
View file @ 97a6d740
...@@ -219,7 +219,7 @@ class GNNExplainer(nn.Module): ...@@ -219,7 +219,7 @@ class GNNExplainer(nn.Module):
>>> new_center >>> new_center
tensor([1]) tensor([1])
>>> sg.num_edges() >>> sg.num_edges()
26 12
>>> # Old IDs of the nodes in the subgraph >>> # Old IDs of the nodes in the subgraph
>>> sg.ndata[dgl.NID] >>> sg.ndata[dgl.NID]
tensor([ 9, 10, 11, 12]) tensor([ 9, 10, 11, 12])
...@@ -229,9 +229,8 @@ class GNNExplainer(nn.Module): ...@@ -229,9 +229,8 @@ class GNNExplainer(nn.Module):
>>> feat_mask >>> feat_mask
tensor([0.2638, 0.2738, 0.3039, ..., 0.2794, 0.2643, 0.2733]) tensor([0.2638, 0.2738, 0.3039, ..., 0.2794, 0.2643, 0.2733])
>>> edge_mask >>> edge_mask
tensor([0.8291, 0.2065, 0.1379, 0.2265, 0.8618, 0.7038, 0.2094, 0.8847, 0.2157, tensor([0.0937, 0.1496, 0.8287, 0.8132, 0.8825, 0.8515, 0.8146, 0.0915, 0.1145,
0.6595, 0.1906, 0.8184, 0.2033, 0.7211, 0.1279, 0.1668, 0.1441, 0.8571, 0.9011, 0.1311, 0.8437])
0.1903, 0.1125, 0.8235, 0.1913, 0.5834, 0.2248, 0.8345, 0.9270])
""" """
self.model.eval() self.model.eval()
num_nodes = graph.num_nodes() num_nodes = graph.num_nodes()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment