| 66 | |
| 67 | |
| 68 | class Dataload(data.Dataset): |
| 69 | def __init__(self, Adj, Node): |
| 70 | self.Adj = Adj |
| 71 | self.Node = Node |
| 72 | |
| 73 | def __getitem__(self, index): |
| 74 | return index |
| 75 | # adj_batch = self.Adj[index] |
| 76 | # adj_mat = adj_batch[index] |
| 77 | # b_mat = torch.ones_like(adj_batch) |
| 78 | # b_mat[adj_batch != 0] = self.Beta |
| 79 | # return adj_batch, adj_mat, b_mat |
| 80 | |
| 81 | def __len__(self): |
| 82 | return self.Node |
| 83 | |
| 84 | |
| 85 | def get_adj(g): |