Fix README.md
This commit is contained in:
parent
2fed8f3b37
commit
5127ba2a27
|
@ -9,6 +9,7 @@ data = dataset[0]
|
||||||
|
|
||||||
import torch.nn.functional as F
|
import torch.nn.functional as F
|
||||||
from torch_geometric.nn import GCNConv, global_mean_pool
|
from torch_geometric.nn import GCNConv, global_mean_pool
|
||||||
|
from eixgnn.eixgnn import EiXGNN
|
||||||
|
|
||||||
class GCN(torch.nn.Module):
|
class GCN(torch.nn.Module):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
|
|
Loading…
Reference in New Issue