| 51 | |
| 52 | # Construct feed dictionary |
| 53 | def construct_feed_dict(pkl, placeholders): |
| 54 | coord = pkl[0] |
| 55 | pool_idx = pkl[4] |
| 56 | faces = pkl[5] |
| 57 | lape_idx = pkl[7] |
| 58 | edges = [] |
| 59 | for i in range(1,4): |
| 60 | adj = pkl[i][1] |
| 61 | edges.append(adj[0]) |
| 62 | |
| 63 | feed_dict = dict() |
| 64 | feed_dict.update({placeholders['features']: coord}) |
| 65 | feed_dict.update({placeholders['edges'][i]: edges[i] for i in range(len(edges))}) |
| 66 | feed_dict.update({placeholders['faces'][i]: faces[i] for i in range(len(faces))}) |
| 67 | feed_dict.update({placeholders['pool_idx'][i]: pool_idx[i] for i in range(len(pool_idx))}) |
| 68 | feed_dict.update({placeholders['lape_idx'][i]: lape_idx[i] for i in range(len(lape_idx))}) |
| 69 | feed_dict.update({placeholders['support1'][i]: pkl[1][i] for i in range(len(pkl[1]))}) |
| 70 | feed_dict.update({placeholders['support2'][i]: pkl[2][i] for i in range(len(pkl[2]))}) |
| 71 | feed_dict.update({placeholders['support3'][i]: pkl[3][i] for i in range(len(pkl[3]))}) |
| 72 | return feed_dict |
| 73 | |
| 74 | def f_score(label, predict, dist_label, dist_pred, threshold): |
| 75 | num_label = label.shape[0] |