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