Add train_idx/val_idx/test_idx as dataset attributes according to corresponding mask.
(self)
| 197 | return 1 |
| 198 | |
| 199 | def _set_split_index(self): |
| 200 | """Add train_idx/val_idx/test_idx as dataset attributes according to corresponding mask.""" |
| 201 | ndata = self.g.nodes[self.target_ntype].data |
| 202 | self.train_idx = F.nonzero_1d(ndata["train_mask"]) |
| 203 | self.val_idx = F.nonzero_1d(ndata["val_mask"]) |
| 204 | self.test_idx = F.nonzero_1d(ndata["test_mask"]) |
| 205 | |
| 206 | |
| 207 | def negative_sample(g, num_samples): |