(self, in_dims=16, embedding_dims=16, dropout=0)
| 235 | class SparseGraphConvolution(nn.Module): |
| 236 | |
| 237 | def __init__(self, in_dims=16, embedding_dims=16, dropout=0): |
| 238 | super(SparseGraphConvolution, self).__init__() |
| 239 | |
| 240 | self.dropout = dropout |
| 241 | |
| 242 | self.spatial_temporal_sparse_gcn = nn.ModuleList() |
| 243 | self.temporal_spatial_sparse_gcn = nn.ModuleList() |
| 244 | |
| 245 | self.spatial_temporal_sparse_gcn.append(GraphConvolution(in_dims, embedding_dims)) |
| 246 | self.spatial_temporal_sparse_gcn.append(GraphConvolution(embedding_dims, embedding_dims)) |
| 247 | |
| 248 | self.temporal_spatial_sparse_gcn.append(GraphConvolution(in_dims, embedding_dims)) |
| 249 | self.temporal_spatial_sparse_gcn.append(GraphConvolution(embedding_dims, embedding_dims)) |
| 250 | |
| 251 | def forward(self, graph, normalized_spatial_adjacency_matrix, normalized_temporal_adjacency_matrix): |
| 252 |
nothing calls this directly
no test coverage detected