(data_dir)
| 90 | return graph #edge_index, vertex_position |
| 91 | |
| 92 | def load_embedding(data_dir): |
| 93 | # embedding_filename = '/embeddings.pickle' |
| 94 | with open(data_dir, 'rb') as f: |
| 95 | embeddings = pickle.load(f) |
| 96 | return embeddings |
| 97 | |
| 98 | |
| 99 | def evaluate(): |