Skip to content
Snippets Groups Projects
Commit 2941cf06 authored by Mateusz Klimaszewski's avatar Mateusz Klimaszewski
Browse files

Add deps sorting.

parent c082b650
2 merge requests!37Release 1.0.4.,!36Release 1.0.4
...@@ -74,8 +74,11 @@ def graph_and_tree_merge(tree_arc_scores, ...@@ -74,8 +74,11 @@ def graph_and_tree_merge(tree_arc_scores,
parse_graph[d] = sorted(parse_graph[d]) parse_graph[d] = sorted(parse_graph[d])
for i, g in enumerate(parse_graph): for i, g in enumerate(parse_graph):
heads = [x[0] for x in g] heads = np.array([x[0] for x in g])
rels = [x[1] for x in g] rels = np.array([x[1] for x in g])
indices = rels.argsort()
heads = heads[indices].tolist()
rels = rels[indices].tolist()
deps = '|'.join(f'{h}:{r}' for h, r in zip(heads, rels)) deps = '|'.join(f'{h}:{r}' for h, r in zip(heads, rels))
tokens[i - 1]["deps"] = deps tokens[i - 1]["deps"] = deps
return return
......
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