You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

util.py 23 kB

5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664
  1. #!/usr/bin/env python3
  2. # -*- coding: utf-8 -*-
  3. """
  4. Created on Tue Mar 31 17:06:22 2020
  5. @author: ljia
  6. """
  7. import numpy as np
  8. from itertools import combinations
  9. import multiprocessing
  10. from multiprocessing import Pool
  11. from functools import partial
  12. import sys
  13. from tqdm import tqdm
  14. import networkx as nx
  15. from gklearn.ged.env import GEDEnv
  16. def compute_ged(g1, g2, options):
  17. from gklearn.gedlib import librariesImport, gedlibpy
  18. ged_env = gedlibpy.GEDEnv()
  19. ged_env.set_edit_cost(options['edit_cost'], edit_cost_constant=options['edit_cost_constants'])
  20. ged_env.add_nx_graph(g1, '')
  21. ged_env.add_nx_graph(g2, '')
  22. listID = ged_env.get_all_graph_ids()
  23. ged_env.init(init_type=options['init_option'])
  24. ged_env.set_method(options['method'], ged_options_to_string(options))
  25. ged_env.init_method()
  26. g = listID[0]
  27. h = listID[1]
  28. ged_env.run_method(g, h)
  29. pi_forward = ged_env.get_forward_map(g, h)
  30. pi_backward = ged_env.get_backward_map(g, h)
  31. upper = ged_env.get_upper_bound(g, h)
  32. dis = upper
  33. # make the map label correct (label remove map as np.inf)
  34. nodes1 = [n for n in g1.nodes()]
  35. nodes2 = [n for n in g2.nodes()]
  36. nb1 = nx.number_of_nodes(g1)
  37. nb2 = nx.number_of_nodes(g2)
  38. pi_forward = [nodes2[pi] if pi < nb2 else np.inf for pi in pi_forward]
  39. pi_backward = [nodes1[pi] if pi < nb1 else np.inf for pi in pi_backward]
  40. # print(pi_forward)
  41. return dis, pi_forward, pi_backward
  42. def pairwise_ged(g1, g2, options={}, sort=True, repeats=1, parallel=False, verbose=True):
  43. from gklearn.gedlib import librariesImport, gedlibpy
  44. ged_env = gedlibpy.GEDEnv()
  45. ged_env.set_edit_cost(options['edit_cost'], edit_cost_constant=options['edit_cost_constants'])
  46. ged_env.add_nx_graph(g1, '')
  47. ged_env.add_nx_graph(g2, '')
  48. listID = ged_env.get_all_graph_ids()
  49. ged_env.init(init_option=(options['init_option'] if 'init_option' in options else 'EAGER_WITHOUT_SHUFFLED_COPIES'))
  50. ged_env.set_method(options['method'], ged_options_to_string(options))
  51. ged_env.init_method()
  52. g = listID[0]
  53. h = listID[1]
  54. dis_min = np.inf
  55. for i in range(0, repeats):
  56. ged_env.run_method(g, h)
  57. upper = ged_env.get_upper_bound(g, h)
  58. dis = upper
  59. if dis < dis_min:
  60. dis_min = dis
  61. pi_forward = ged_env.get_forward_map(g, h)
  62. pi_backward = ged_env.get_backward_map(g, h)
  63. # lower = ged_env.get_lower_bound(g, h)
  64. # make the map label correct (label remove map as np.inf)
  65. nodes1 = [n for n in g1.nodes()]
  66. nodes2 = [n for n in g2.nodes()]
  67. nb1 = nx.number_of_nodes(g1)
  68. nb2 = nx.number_of_nodes(g2)
  69. pi_forward = [nodes2[pi] if pi < nb2 else np.inf for pi in pi_forward]
  70. pi_backward = [nodes1[pi] if pi < nb1 else np.inf for pi in pi_backward]
  71. # print(pi_forward)
  72. return dis, pi_forward, pi_backward
  73. def compute_geds_cml(graphs, options={}, sort=True, parallel=False, verbose=True):
  74. # initialize ged env.
  75. ged_env = GEDEnv()
  76. ged_env.set_edit_cost(options['edit_cost'], edit_cost_constants=options['edit_cost_constants'])
  77. for g in graphs:
  78. ged_env.add_nx_graph(g, '')
  79. listID = ged_env.get_all_graph_ids()
  80. node_labels = ged_env.get_all_node_labels()
  81. edge_labels = ged_env.get_all_edge_labels()
  82. node_label_costs = label_costs_to_matrix(options['node_label_costs'], len(node_labels)) if 'node_label_costs' in options else None
  83. edge_label_costs = label_costs_to_matrix(options['edge_label_costs'], len(edge_labels)) if 'edge_label_costs' in options else None
  84. ged_env.set_label_costs(node_label_costs, edge_label_costs)
  85. ged_env.init(init_type=options['init_option'])
  86. if parallel:
  87. options['threads'] = 1
  88. ged_env.set_method(options['method'], options)
  89. ged_env.init_method()
  90. # compute ged.
  91. # options used to compute numbers of edit operations.
  92. if node_label_costs is None and edge_label_costs is None:
  93. neo_options = {'edit_cost': options['edit_cost'],
  94. 'is_cml': False,
  95. 'node_labels': options['node_labels'], 'edge_labels': options['edge_labels'],
  96. 'node_attrs': options['node_attrs'], 'edge_attrs': options['edge_attrs']}
  97. else:
  98. neo_options = {'edit_cost': options['edit_cost'],
  99. 'is_cml': True,
  100. 'node_labels': node_labels,
  101. 'edge_labels': edge_labels}
  102. ged_mat = np.zeros((len(graphs), len(graphs)))
  103. if parallel:
  104. len_itr = int(len(graphs) * (len(graphs) - 1) / 2)
  105. ged_vec = [0 for i in range(len_itr)]
  106. n_edit_operations = [0 for i in range(len_itr)]
  107. itr = combinations(range(0, len(graphs)), 2)
  108. n_jobs = multiprocessing.cpu_count()
  109. if len_itr < 100 * n_jobs:
  110. chunksize = int(len_itr / n_jobs) + 1
  111. else:
  112. chunksize = 100
  113. def init_worker(graphs_toshare, ged_env_toshare, listID_toshare):
  114. global G_graphs, G_ged_env, G_listID
  115. G_graphs = graphs_toshare
  116. G_ged_env = ged_env_toshare
  117. G_listID = listID_toshare
  118. do_partial = partial(_wrapper_compute_ged_parallel, neo_options, sort)
  119. pool = Pool(processes=n_jobs, initializer=init_worker, initargs=(graphs, ged_env, listID))
  120. if verbose:
  121. iterator = tqdm(pool.imap_unordered(do_partial, itr, chunksize),
  122. desc='computing GEDs', file=sys.stdout)
  123. else:
  124. iterator = pool.imap_unordered(do_partial, itr, chunksize)
  125. # iterator = pool.imap_unordered(do_partial, itr, chunksize)
  126. for i, j, dis, n_eo_tmp in iterator:
  127. idx_itr = int(len(graphs) * i + j - (i + 1) * (i + 2) / 2)
  128. ged_vec[idx_itr] = dis
  129. ged_mat[i][j] = dis
  130. ged_mat[j][i] = dis
  131. n_edit_operations[idx_itr] = n_eo_tmp
  132. # print('\n-------------------------------------------')
  133. # print(i, j, idx_itr, dis)
  134. pool.close()
  135. pool.join()
  136. else:
  137. ged_vec = []
  138. n_edit_operations = []
  139. if verbose:
  140. iterator = tqdm(range(len(graphs)), desc='computing GEDs', file=sys.stdout)
  141. else:
  142. iterator = range(len(graphs))
  143. for i in iterator:
  144. # for i in range(len(graphs)):
  145. for j in range(i + 1, len(graphs)):
  146. if nx.number_of_nodes(graphs[i]) <= nx.number_of_nodes(graphs[j]) or not sort:
  147. dis, pi_forward, pi_backward = _compute_ged(ged_env, listID[i], listID[j], graphs[i], graphs[j])
  148. else:
  149. dis, pi_backward, pi_forward = _compute_ged(ged_env, listID[j], listID[i], graphs[j], graphs[i])
  150. ged_vec.append(dis)
  151. ged_mat[i][j] = dis
  152. ged_mat[j][i] = dis
  153. n_eo_tmp = get_nb_edit_operations(graphs[i], graphs[j], pi_forward, pi_backward, **neo_options)
  154. n_edit_operations.append(n_eo_tmp)
  155. return ged_vec, ged_mat, n_edit_operations
  156. def compute_geds(graphs, options={}, sort=True, repeats=1, parallel=False, verbose=True):
  157. from gklearn.gedlib import librariesImport, gedlibpy
  158. # initialize ged env.
  159. ged_env = gedlibpy.GEDEnv()
  160. ged_env.set_edit_cost(options['edit_cost'], edit_cost_constant=options['edit_cost_constants'])
  161. for g in graphs:
  162. ged_env.add_nx_graph(g, '')
  163. listID = ged_env.get_all_graph_ids()
  164. ged_env.init()
  165. if parallel:
  166. options['threads'] = 1
  167. ged_env.set_method(options['method'], ged_options_to_string(options))
  168. ged_env.init_method()
  169. # compute ged.
  170. neo_options = {'edit_cost': options['edit_cost'],
  171. 'node_labels': options['node_labels'], 'edge_labels': options['edge_labels'],
  172. 'node_attrs': options['node_attrs'], 'edge_attrs': options['edge_attrs']}
  173. ged_mat = np.zeros((len(graphs), len(graphs)))
  174. if parallel:
  175. len_itr = int(len(graphs) * (len(graphs) - 1) / 2)
  176. ged_vec = [0 for i in range(len_itr)]
  177. n_edit_operations = [0 for i in range(len_itr)]
  178. itr = combinations(range(0, len(graphs)), 2)
  179. n_jobs = multiprocessing.cpu_count()
  180. if len_itr < 100 * n_jobs:
  181. chunksize = int(len_itr / n_jobs) + 1
  182. else:
  183. chunksize = 100
  184. def init_worker(graphs_toshare, ged_env_toshare, listID_toshare):
  185. global G_graphs, G_ged_env, G_listID
  186. G_graphs = graphs_toshare
  187. G_ged_env = ged_env_toshare
  188. G_listID = listID_toshare
  189. do_partial = partial(_wrapper_compute_ged_parallel, neo_options, sort, repeats)
  190. pool = Pool(processes=n_jobs, initializer=init_worker, initargs=(graphs, ged_env, listID))
  191. if verbose:
  192. iterator = tqdm(pool.imap_unordered(do_partial, itr, chunksize),
  193. desc='computing GEDs', file=sys.stdout)
  194. else:
  195. iterator = pool.imap_unordered(do_partial, itr, chunksize)
  196. # iterator = pool.imap_unordered(do_partial, itr, chunksize)
  197. for i, j, dis, n_eo_tmp in iterator:
  198. idx_itr = int(len(graphs) * i + j - (i + 1) * (i + 2) / 2)
  199. ged_vec[idx_itr] = dis
  200. ged_mat[i][j] = dis
  201. ged_mat[j][i] = dis
  202. n_edit_operations[idx_itr] = n_eo_tmp
  203. # print('\n-------------------------------------------')
  204. # print(i, j, idx_itr, dis)
  205. pool.close()
  206. pool.join()
  207. else:
  208. ged_vec = []
  209. n_edit_operations = []
  210. if verbose:
  211. iterator = tqdm(range(len(graphs)), desc='computing GEDs', file=sys.stdout)
  212. else:
  213. iterator = range(len(graphs))
  214. for i in iterator:
  215. # for i in range(len(graphs)):
  216. for j in range(i + 1, len(graphs)):
  217. if nx.number_of_nodes(graphs[i]) <= nx.number_of_nodes(graphs[j]) or not sort:
  218. dis, pi_forward, pi_backward = _compute_ged(ged_env, listID[i], listID[j], graphs[i], graphs[j], repeats)
  219. else:
  220. dis, pi_backward, pi_forward = _compute_ged(ged_env, listID[j], listID[i], graphs[j], graphs[i], repeats)
  221. ged_vec.append(dis)
  222. ged_mat[i][j] = dis
  223. ged_mat[j][i] = dis
  224. n_eo_tmp = get_nb_edit_operations(graphs[i], graphs[j], pi_forward, pi_backward, **neo_options)
  225. n_edit_operations.append(n_eo_tmp)
  226. return ged_vec, ged_mat, n_edit_operations
  227. def _wrapper_compute_ged_parallel(options, sort, repeats, itr):
  228. i = itr[0]
  229. j = itr[1]
  230. dis, n_eo_tmp = _compute_ged_parallel(G_ged_env, G_listID[i], G_listID[j], G_graphs[i], G_graphs[j], options, sort, repeats)
  231. return i, j, dis, n_eo_tmp
  232. def _compute_ged_parallel(env, gid1, gid2, g1, g2, options, sort, repeats):
  233. if nx.number_of_nodes(g1) <= nx.number_of_nodes(g2) or not sort:
  234. dis, pi_forward, pi_backward = _compute_ged(env, gid1, gid2, g1, g2, repeats)
  235. else:
  236. dis, pi_backward, pi_forward = _compute_ged(env, gid2, gid1, g2, g1, repeats)
  237. n_eo_tmp = get_nb_edit_operations(g1, g2, pi_forward, pi_backward, **options) # [0,0,0,0,0,0]
  238. return dis, n_eo_tmp
  239. def _compute_ged(env, gid1, gid2, g1, g2, repeats):
  240. dis_min = np.inf # @todo: maybe compare distance and then do others (faster).
  241. for i in range(0, repeats):
  242. env.run_method(gid1, gid2)
  243. pi_forward = env.get_forward_map(gid1, gid2)
  244. pi_backward = env.get_backward_map(gid1, gid2)
  245. upper = env.get_upper_bound(gid1, gid2)
  246. dis = upper
  247. # make the map label correct (label remove map as np.inf)
  248. nodes1 = [n for n in g1.nodes()]
  249. nodes2 = [n for n in g2.nodes()]
  250. nb1 = nx.number_of_nodes(g1)
  251. nb2 = nx.number_of_nodes(g2)
  252. pi_forward = [nodes2[pi] if pi < nb2 else np.inf for pi in pi_forward]
  253. pi_backward = [nodes1[pi] if pi < nb1 else np.inf for pi in pi_backward]
  254. if dis < dis_min:
  255. dis_min = dis
  256. pi_forward_min = pi_forward
  257. pi_backward_min = pi_backward
  258. return dis_min, pi_forward_min, pi_backward_min
  259. def label_costs_to_matrix(costs, nb_labels):
  260. """Reform a label cost vector to a matrix.
  261. Parameters
  262. ----------
  263. costs : numpy.array
  264. The vector containing costs between labels, in the order of node insertion costs, node deletion costs, node substitition costs, edge insertion costs, edge deletion costs, edge substitition costs.
  265. nb_labels : integer
  266. Number of labels.
  267. Returns
  268. -------
  269. cost_matrix : numpy.array.
  270. The reformed label cost matrix of size (nb_labels, nb_labels). Each row/column of cost_matrix corresponds to a label, and the first label is the dummy label. This is the same setting as in GEDData.
  271. """
  272. # Initialize label cost matrix.
  273. cost_matrix = np.zeros((nb_labels + 1, nb_labels + 1))
  274. i = 0
  275. # Costs of insertions.
  276. for col in range(1, nb_labels + 1):
  277. cost_matrix[0, col] = costs[i]
  278. i += 1
  279. # Costs of deletions.
  280. for row in range(1, nb_labels + 1):
  281. cost_matrix[row, 0] = costs[i]
  282. i += 1
  283. # Costs of substitutions.
  284. for row in range(1, nb_labels + 1):
  285. for col in range(row + 1, nb_labels + 1):
  286. cost_matrix[row, col] = costs[i]
  287. cost_matrix[col, row] = costs[i]
  288. i += 1
  289. return cost_matrix
  290. def get_nb_edit_operations(g1, g2, forward_map, backward_map, edit_cost=None, is_cml=False, **kwargs):
  291. if is_cml:
  292. if edit_cost == 'CONSTANT':
  293. node_labels = kwargs.get('node_labels', [])
  294. edge_labels = kwargs.get('edge_labels', [])
  295. return get_nb_edit_operations_symbolic_cml(g1, g2, forward_map, backward_map,
  296. node_labels=node_labels, edge_labels=edge_labels)
  297. else:
  298. raise Exception('Edit cost "', edit_cost, '" is not supported.')
  299. else:
  300. if edit_cost == 'LETTER' or edit_cost == 'LETTER2':
  301. return get_nb_edit_operations_letter(g1, g2, forward_map, backward_map)
  302. elif edit_cost == 'NON_SYMBOLIC':
  303. node_attrs = kwargs.get('node_attrs', [])
  304. edge_attrs = kwargs.get('edge_attrs', [])
  305. return get_nb_edit_operations_nonsymbolic(g1, g2, forward_map, backward_map,
  306. node_attrs=node_attrs, edge_attrs=edge_attrs)
  307. elif edit_cost == 'CONSTANT':
  308. node_labels = kwargs.get('node_labels', [])
  309. edge_labels = kwargs.get('edge_labels', [])
  310. return get_nb_edit_operations_symbolic(g1, g2, forward_map, backward_map,
  311. node_labels=node_labels, edge_labels=edge_labels)
  312. else:
  313. return get_nb_edit_operations_symbolic(g1, g2, forward_map, backward_map)
  314. def get_nb_edit_operations_symbolic_cml(g1, g2, forward_map, backward_map,
  315. node_labels=[], edge_labels=[]):
  316. """Compute times that edit operations are used in an edit path for symbolic-labeled graphs, where the costs are different for each pair of nodes.
  317. Returns
  318. -------
  319. list
  320. A vector of numbers of times that costs bewteen labels are used in an edit path, formed in the order of node insertion costs, node deletion costs, node substitition costs, edge insertion costs, edge deletion costs, edge substitition costs. The dummy label is the first label, and the self label costs are not included.
  321. """
  322. # Initialize.
  323. nb_ops_node = np.zeros((1 + len(node_labels), 1 + len(node_labels)))
  324. nb_ops_edge = np.zeros((1 + len(edge_labels), 1 + len(edge_labels)))
  325. # For nodes.
  326. nodes1 = [n for n in g1.nodes()]
  327. for i, map_i in enumerate(forward_map):
  328. label1 = tuple(g1.nodes[nodes1[i]].items()) # @todo: order and faster
  329. idx_label1 = node_labels.index(label1) # @todo: faster
  330. if map_i == np.inf: # deletions.
  331. nb_ops_node[idx_label1 + 1, 0] += 1
  332. else: # substitutions.
  333. label2 = tuple(g2.nodes[map_i].items())
  334. if label1 != label2:
  335. idx_label2 = node_labels.index(label2) # @todo: faster
  336. nb_ops_node[idx_label1 + 1, idx_label2 + 1] += 1
  337. # insertions.
  338. nodes2 = [n for n in g2.nodes()]
  339. for i, map_i in enumerate(backward_map):
  340. if map_i == np.inf:
  341. label = tuple(g2.nodes[nodes2[i]].items())
  342. idx_label = node_labels.index(label) # @todo: faster
  343. nb_ops_node[0, idx_label + 1] += 1
  344. # For edges.
  345. edges1 = [e for e in g1.edges()]
  346. edges2_marked = []
  347. for nf1, nt1 in edges1:
  348. label1 = tuple(g1.edges[(nf1, nt1)].items())
  349. idx_label1 = edge_labels.index(label1) # @todo: faster
  350. idxf1 = nodes1.index(nf1) # @todo: faster
  351. idxt1 = nodes1.index(nt1) # @todo: faster
  352. # At least one of the nodes is removed, thus the edge is removed.
  353. if forward_map[idxf1] == np.inf or forward_map[idxt1] == np.inf:
  354. nb_ops_edge[idx_label1 + 1, 0] += 1
  355. # corresponding edge is in g2.
  356. else:
  357. nf2, nt2 = forward_map[idxf1], forward_map[idxt1]
  358. if (nf2, nt2) in g2.edges():
  359. edges2_marked.append((nf2, nt2))
  360. # If edge labels are different.
  361. label2 = tuple(g2.edges[(nf2, nt2)].items())
  362. if label1 != label2:
  363. idx_label2 = edge_labels.index(label2) # @todo: faster
  364. nb_ops_edge[idx_label1 + 1, idx_label2 + 1] += 1
  365. # Switch nf2 and nt2, for directed graphs.
  366. elif (nt2, nf2) in g2.edges():
  367. edges2_marked.append((nt2, nf2))
  368. # If edge labels are different.
  369. label2 = tuple(g2.edges[(nt2, nf2)].items())
  370. if label1 != label2:
  371. idx_label2 = edge_labels.index(label2) # @todo: faster
  372. nb_ops_edge[idx_label1 + 1, idx_label2 + 1] += 1
  373. # Corresponding nodes are in g2, however the edge is removed.
  374. else:
  375. nb_ops_edge[idx_label1 + 1, 0] += 1
  376. # insertions.
  377. for nt, nf in g2.edges():
  378. if (nt, nf) not in edges2_marked and (nf, nt) not in edges2_marked: # @todo: for directed.
  379. label = tuple(g2.edges[(nt, nf)].items())
  380. idx_label = edge_labels.index(label) # @todo: faster
  381. nb_ops_edge[0, idx_label + 1] += 1
  382. # Reform the numbers of edit oeprations into a vector.
  383. nb_eo_vector = []
  384. # node insertion.
  385. for i in range(1, len(nb_ops_node)):
  386. nb_eo_vector.append(nb_ops_node[0, i])
  387. # node deletion.
  388. for i in range(1, len(nb_ops_node)):
  389. nb_eo_vector.append(nb_ops_node[i, 0])
  390. # node substitution.
  391. for i in range(1, len(nb_ops_node)):
  392. for j in range(i + 1, len(nb_ops_node)):
  393. nb_eo_vector.append(nb_ops_node[i, j])
  394. # edge insertion.
  395. for i in range(1, len(nb_ops_edge)):
  396. nb_eo_vector.append(nb_ops_edge[0, i])
  397. # edge deletion.
  398. for i in range(1, len(nb_ops_edge)):
  399. nb_eo_vector.append(nb_ops_edge[i, 0])
  400. # edge substitution.
  401. for i in range(1, len(nb_ops_edge)):
  402. for j in range(i + 1, len(nb_ops_edge)):
  403. nb_eo_vector.append(nb_ops_edge[i, j])
  404. return nb_eo_vector
  405. def get_nb_edit_operations_symbolic(g1, g2, forward_map, backward_map,
  406. node_labels=[], edge_labels=[]):
  407. """Compute the number of each edit operations for symbolic-labeled graphs.
  408. """
  409. n_vi = 0
  410. n_vr = 0
  411. n_vs = 0
  412. n_ei = 0
  413. n_er = 0
  414. n_es = 0
  415. nodes1 = [n for n in g1.nodes()]
  416. for i, map_i in enumerate(forward_map):
  417. if map_i == np.inf:
  418. n_vr += 1
  419. else:
  420. for nl in node_labels:
  421. label1 = g1.nodes[nodes1[i]][nl]
  422. label2 = g2.nodes[map_i][nl]
  423. if label1 != label2:
  424. n_vs += 1
  425. break
  426. for map_i in backward_map:
  427. if map_i == np.inf:
  428. n_vi += 1
  429. # idx_nodes1 = range(0, len(node1))
  430. edges1 = [e for e in g1.edges()]
  431. nb_edges2_cnted = 0
  432. for n1, n2 in edges1:
  433. idx1 = nodes1.index(n1)
  434. idx2 = nodes1.index(n2)
  435. # one of the nodes is removed, thus the edge is removed.
  436. if forward_map[idx1] == np.inf or forward_map[idx2] == np.inf:
  437. n_er += 1
  438. # corresponding edge is in g2.
  439. elif (forward_map[idx1], forward_map[idx2]) in g2.edges():
  440. nb_edges2_cnted += 1
  441. # edge labels are different.
  442. for el in edge_labels:
  443. label1 = g2.edges[((forward_map[idx1], forward_map[idx2]))][el]
  444. label2 = g1.edges[(n1, n2)][el]
  445. if label1 != label2:
  446. n_es += 1
  447. break
  448. elif (forward_map[idx2], forward_map[idx1]) in g2.edges():
  449. nb_edges2_cnted += 1
  450. # edge labels are different.
  451. for el in edge_labels:
  452. label1 = g2.edges[((forward_map[idx2], forward_map[idx1]))][el]
  453. label2 = g1.edges[(n1, n2)][el]
  454. if label1 != label2:
  455. n_es += 1
  456. break
  457. # corresponding nodes are in g2, however the edge is removed.
  458. else:
  459. n_er += 1
  460. n_ei = nx.number_of_edges(g2) - nb_edges2_cnted
  461. return n_vi, n_vr, n_vs, n_ei, n_er, n_es
  462. def get_nb_edit_operations_letter(g1, g2, forward_map, backward_map):
  463. """Compute the number of each edit operations.
  464. """
  465. n_vi = 0
  466. n_vr = 0
  467. n_vs = 0
  468. sod_vs = 0
  469. n_ei = 0
  470. n_er = 0
  471. nodes1 = [n for n in g1.nodes()]
  472. for i, map_i in enumerate(forward_map):
  473. if map_i == np.inf:
  474. n_vr += 1
  475. else:
  476. n_vs += 1
  477. diff_x = float(g1.nodes[nodes1[i]]['x']) - float(g2.nodes[map_i]['x'])
  478. diff_y = float(g1.nodes[nodes1[i]]['y']) - float(g2.nodes[map_i]['y'])
  479. sod_vs += np.sqrt(np.square(diff_x) + np.square(diff_y))
  480. for map_i in backward_map:
  481. if map_i == np.inf:
  482. n_vi += 1
  483. # idx_nodes1 = range(0, len(node1))
  484. edges1 = [e for e in g1.edges()]
  485. nb_edges2_cnted = 0
  486. for n1, n2 in edges1:
  487. idx1 = nodes1.index(n1)
  488. idx2 = nodes1.index(n2)
  489. # one of the nodes is removed, thus the edge is removed.
  490. if forward_map[idx1] == np.inf or forward_map[idx2] == np.inf:
  491. n_er += 1
  492. # corresponding edge is in g2. Edge label is not considered.
  493. elif (forward_map[idx1], forward_map[idx2]) in g2.edges() or \
  494. (forward_map[idx2], forward_map[idx1]) in g2.edges():
  495. nb_edges2_cnted += 1
  496. # corresponding nodes are in g2, however the edge is removed.
  497. else:
  498. n_er += 1
  499. n_ei = nx.number_of_edges(g2) - nb_edges2_cnted
  500. return n_vi, n_vr, n_vs, sod_vs, n_ei, n_er
  501. def get_nb_edit_operations_nonsymbolic(g1, g2, forward_map, backward_map,
  502. node_attrs=[], edge_attrs=[]):
  503. """Compute the number of each edit operations.
  504. """
  505. n_vi = 0
  506. n_vr = 0
  507. n_vs = 0
  508. sod_vs = 0
  509. n_ei = 0
  510. n_er = 0
  511. n_es = 0
  512. sod_es = 0
  513. nodes1 = [n for n in g1.nodes()]
  514. for i, map_i in enumerate(forward_map):
  515. if map_i == np.inf:
  516. n_vr += 1
  517. else:
  518. n_vs += 1
  519. sum_squares = 0
  520. for a_name in node_attrs:
  521. diff = float(g1.nodes[nodes1[i]][a_name]) - float(g2.nodes[map_i][a_name])
  522. sum_squares += np.square(diff)
  523. sod_vs += np.sqrt(sum_squares)
  524. for map_i in backward_map:
  525. if map_i == np.inf:
  526. n_vi += 1
  527. # idx_nodes1 = range(0, len(node1))
  528. edges1 = [e for e in g1.edges()]
  529. for n1, n2 in edges1:
  530. idx1 = nodes1.index(n1)
  531. idx2 = nodes1.index(n2)
  532. n1_g2 = forward_map[idx1]
  533. n2_g2 = forward_map[idx2]
  534. # one of the nodes is removed, thus the edge is removed.
  535. if n1_g2 == np.inf or n2_g2 == np.inf:
  536. n_er += 1
  537. # corresponding edge is in g2.
  538. elif (n1_g2, n2_g2) in g2.edges():
  539. n_es += 1
  540. sum_squares = 0
  541. for a_name in edge_attrs:
  542. diff = float(g1.edges[n1, n2][a_name]) - float(g2.edges[n1_g2, n2_g2][a_name])
  543. sum_squares += np.square(diff)
  544. sod_es += np.sqrt(sum_squares)
  545. elif (n2_g2, n1_g2) in g2.edges():
  546. n_es += 1
  547. sum_squares = 0
  548. for a_name in edge_attrs:
  549. diff = float(g1.edges[n2, n1][a_name]) - float(g2.edges[n2_g2, n1_g2][a_name])
  550. sum_squares += np.square(diff)
  551. sod_es += np.sqrt(sum_squares)
  552. # corresponding nodes are in g2, however the edge is removed.
  553. else:
  554. n_er += 1
  555. n_ei = nx.number_of_edges(g2) - n_es
  556. return n_vi, n_vr, sod_vs, n_ei, n_er, sod_es
  557. def ged_options_to_string(options):
  558. opt_str = ' '
  559. for key, val in options.items():
  560. if key == 'initialization_method':
  561. opt_str += '--initialization-method ' + str(val) + ' '
  562. elif key == 'initialization_options':
  563. opt_str += '--initialization-options ' + str(val) + ' '
  564. elif key == 'lower_bound_method':
  565. opt_str += '--lower-bound-method ' + str(val) + ' '
  566. elif key == 'random_substitution_ratio':
  567. opt_str += '--random-substitution-ratio ' + str(val) + ' '
  568. elif key == 'initial_solutions':
  569. opt_str += '--initial-solutions ' + str(val) + ' '
  570. elif key == 'ratio_runs_from_initial_solutions':
  571. opt_str += '--ratio-runs-from-initial-solutions ' + str(val) + ' '
  572. elif key == 'threads':
  573. opt_str += '--threads ' + str(val) + ' '
  574. elif key == 'num_randpost_loops':
  575. opt_str += '--num-randpost-loops ' + str(val) + ' '
  576. elif key == 'max_randpost_retrials':
  577. opt_str += '--maxrandpost-retrials ' + str(val) + ' '
  578. elif key == 'randpost_penalty':
  579. opt_str += '--randpost-penalty ' + str(val) + ' '
  580. elif key == 'randpost_decay':
  581. opt_str += '--randpost-decay ' + str(val) + ' '
  582. elif key == 'log':
  583. opt_str += '--log ' + str(val) + ' '
  584. elif key == 'randomness':
  585. opt_str += '--randomness ' + str(val) + ' '
  586. # if not isinstance(val, list):
  587. # opt_str += '--' + key.replace('_', '-') + ' '
  588. # if val == False:
  589. # val_str = 'FALSE'
  590. # else:
  591. # val_str = str(val)
  592. # opt_str += val_str + ' '
  593. return opt_str

A Python package for graph kernels, graph edit distances and graph pre-image problem.