Construct feed dictionary.
(pkl, placeholders)
| 17 | import numpy as np |
| 18 | |
| 19 | def construct_feed_dict(pkl, placeholders): |
| 20 | """Construct feed dictionary.""" |
| 21 | coord = pkl[0] |
| 22 | pool_idx = pkl[4] |
| 23 | faces = pkl[5] |
| 24 | # laplace = pkl[6] |
| 25 | lape_idx = pkl[7] |
| 26 | |
| 27 | edges = [] |
| 28 | for i in range(1,4): |
| 29 | adj = pkl[i][1] |
| 30 | edges.append(adj[0]) |
| 31 | |
| 32 | feed_dict = dict() |
| 33 | feed_dict.update({placeholders['features']: coord}) |
| 34 | feed_dict.update({placeholders['edges'][i]: edges[i] for i in range(len(edges))}) |
| 35 | feed_dict.update({placeholders['faces'][i]: faces[i] for i in range(len(faces))}) |
| 36 | feed_dict.update({placeholders['pool_idx'][i]: pool_idx[i] for i in range(len(pool_idx))}) |
| 37 | feed_dict.update({placeholders['lape_idx'][i]: lape_idx[i] for i in range(len(lape_idx))}) |
| 38 | feed_dict.update({placeholders['support1'][i]: pkl[1][i] for i in range(len(pkl[1]))}) |
| 39 | feed_dict.update({placeholders['support2'][i]: pkl[2][i] for i in range(len(pkl[2]))}) |
| 40 | feed_dict.update({placeholders['support3'][i]: pkl[3][i] for i in range(len(pkl[3]))}) |
| 41 | return feed_dict |