conv2(x, edge_index, edge_weight) x = global_mean_pool(x, batch) x= `self.linear1(x) x = torch.sigmoid(x) return x 这是GNNexplainer 代码: for batch in test_loader: graphs_batch, labels_batch = batch # Choose the first graph from the batch for explanation graph_to_explain = ...
op = MetaLayer(EdgeModel(), NodeModel(), GlobalModel()) x, edge_attr, u = op(x, edge_index, edge_attr, u, batch) """ def __init__(self, edge_model=None, node_model=None, global_model=None): super(MetaLayer, self).__init__() self.edge_model = edge_model 26 changes: ...
return torch._C._cuda_getDeviceCount() > 0 No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda' /usr/local/lib/python3.8/dist-packages/setuptools/dist.py:717: UserWarning: Usage of dash-separated 'description-file' will not be supported in future versions. Please use the undersco...