Commit a21026c0 authored by Carlos GO's avatar Carlos GO
Browse files

array

parent 1c086b5a
......@@ -145,7 +145,7 @@ array([ 8., 10., 10., 8., 22., 10., 6., 8., 8., 18., 20., 8., 10.,
>>> G = nx.read_gpickle('/data/1jau.nxpickle')
>>> prototoypes = pickle.load('/data/sample_prototoypes.pickle')
>>> graph_embed(G, prototypes)
[1, 0, 2, 4, 1, 4, 5]
array([1., 0., 2., 4., 1., 4., 5.])
```
......
......@@ -204,7 +204,7 @@ def graph_embed(G, prototypes):
for p in prototypes:
ops,_,_ = ged((G,p), source_only=True)
embedding.append(ops.cost)
return embedding
return np.array(embedding)
def full_embed(D, m, DM=None, dist_mat=None, heuristic='spanning'):
"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment