(file_dir, total_label, graph_name, fea_len, train_num)
| 95 | return node_ids, type, label, feature |
| 96 | |
| 97 | def parse_graph_file(file_dir, total_label, graph_name, fea_len, train_num): |
| 98 | content_file = os.path.join(file_dir, "data", graph_name + ".NODE.paper.tab") |
| 99 | cite_file = os.path.join(file_dir, "data", graph_name + ".DIRECTED.cites.tab") |
| 100 | node_map = gen_node_map(cite_file) |
| 101 | edge_src, edge_dst, edge_type = gen_edge(cite_file, node_map, train_num) |
| 102 | node_ids, node_type, node_label, node_feature = gen_node(content_file, node_map, total_label, fea_len, train_num) |
| 103 | return node_ids, node_type, node_label, node_feature, edge_src, edge_dst, edge_type |
nothing calls this directly
no test coverage detected