(cite_file)
| 21 | import numpy as np |
| 22 | |
| 23 | def gen_node_map(cite_file): |
| 24 | node_map = {} |
| 25 | node_index = 0 |
| 26 | with open(os.path.realpath(cite_file), 'r') as f: |
| 27 | for line in f: |
| 28 | nodes = line.strip().split("\t") |
| 29 | if len(nodes) != 2: |
| 30 | continue |
| 31 | for one_node in nodes: |
| 32 | if not node_map.has_key(one_node): |
| 33 | node_map[one_node] = node_index |
| 34 | node_index += 1 |
| 35 | return node_map |
| 36 | |
| 37 | def gen_edge(cite_file, node_map, train_num): |
| 38 | src = [] |