(self, data_dir=reddit_dir, data_type='all')
| 33 | class reddit(DataSet): |
| 34 | |
| 35 | def __init__(self, data_dir=reddit_dir, data_type='all'): |
| 36 | super(reddit, self).__init__(data_dir, data_type) |
| 37 | self.source_url = 'http://snap.stanford.edu/graphsage/reddit.zip' |
| 38 | self.origin_files = ['reddit-class_map.json', |
| 39 | 'reddit-feats.npy', |
| 40 | 'reddit-G.json', |
| 41 | 'reddit-id_map.json', |
| 42 | 'reddit-walks.txt'] |
| 43 | self.max_node_id = 240000 |
| 44 | self.train_node_type = ['train'] |
| 45 | self.train_edge_type = ['train'] |
| 46 | self.total_size = 231443 |
| 47 | self.all_node_type = -1 |
| 48 | self.all_edge_type = ['train', 'train_removed'] |
| 49 | self.id_file = os.path.join(data_dir, 'reddit_test.id') |
| 50 | self.feature_idx = 'feature' |
| 51 | self.feature_dim = 602 |
| 52 | self.label_idx = 'label' |
| 53 | self.label_dim = 1 |
| 54 | self.num_classes = 41 |
| 55 | |
| 56 | def download_data(self, source_url, out_dir): |
| 57 | reddit_zip_dir = os.path.join(out_dir, 'reddit.zip') |
nothing calls this directly
no outgoing calls
no test coverage detected