| 109 | node_name = {"yelp": "review", "amazon": "user"} |
| 110 | |
| 111 | def __init__(self, root, name, transform=None, pre_transform=None, random_seed=717, |
| 112 | train_size=0.7, val_size=0.1, force_reload=False): |
| 113 | |
| 114 | self.name = name |
| 115 | assert self.name in ['yelp', 'amazon'] |
| 116 | |
| 117 | self.url = osp.join(self.url, self.file_urls[self.name]) |
| 118 | self.seed = random_seed |
| 119 | self.train_size = train_size |
| 120 | self.val_size = val_size |
| 121 | |
| 122 | super().__init__(root, transform, pre_transform) |
| 123 | self.load(self.processed_paths[0]) |
| 124 | |
| 125 | @property |
| 126 | def raw_dir(self): |