DIG
DIG copied to clipboard
the methods in xgraph can only explain model in dig.xgraph.models
the methods in xgraph can only explain model in dig.xgraph.models,i try to use it to explain my model which is built by PYG, get the error:
KeyError: 'explain_message'
this is my model:
from torch_geometric.nn import GCNConv
class Net(torch.nn.Module):
def __init__(self):
super().__init__()
self.conv1 = GCNConv(dataset.num_features, 16, normalize=False)
self.conv2 = GCNConv(16, dataset.num_classes, normalize=False)
def forward(self, x, edge_index):
x = F.relu(self.conv1(x, edge_index))
x = F.dropout(x, training=self.training)
x = self.conv2(x, edge_index)
return F.log_softmax(x, dim=1)