From 2941cf0645739c590a0872fbfafb66ffbbaa3481 Mon Sep 17 00:00:00 2001
From: Mateusz Klimaszewski <mk.klimaszewski@gmail.com>
Date: Fri, 7 May 2021 16:07:20 +0200
Subject: [PATCH] Add deps sorting.
---
combo/utils/graph.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/combo/utils/graph.py b/combo/utils/graph.py
index 3995f4f..ed2c1ff 100644
--- a/combo/utils/graph.py
+++ b/combo/utils/graph.py
@@ -74,8 +74,11 @@ def graph_and_tree_merge(tree_arc_scores,
parse_graph[d] = sorted(parse_graph[d])
for i, g in enumerate(parse_graph):
- heads = [x[0] for x in g]
- rels = [x[1] for x in g]
+ heads = np.array([x[0] 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))
tokens[i - 1]["deps"] = deps
return
--
GitLab