(self, in_dim, emb_dim, num_layer, kernel='gcn', drop_ratio=0,
act='relu', norm=None)
| 102 | class GraphCL(nn.Module): |
| 103 | |
| 104 | def __init__(self, in_dim, emb_dim, num_layer, kernel='gcn', drop_ratio=0, |
| 105 | act='relu', norm=None): |
| 106 | super().__init__() |
| 107 | |
| 108 | self.lin_emb_dim = emb_dim * num_layer |
| 109 | self.encoder = Encoder(in_dim, emb_dim, num_layer, kernel, drop_ratio, act, norm) |
| 110 | self.proj_head = LinearPred(self.lin_emb_dim, emb_dim, emb_dim, 2) |
| 111 | |
| 112 | |
| 113 | def forward(self, x, edge_index, edge_weigt=None, batch=None): |
nothing calls this directly
no test coverage detected