|
|
|
|
|
|
|
from data.parser.to_mrp.abstract_parser import AbstractParser |
|
|
|
|
|
class SequentialParser(AbstractParser): |
|
def parse(self, prediction): |
|
output = {} |
|
|
|
output["id"] = self.dataset.id_field.vocab.itos[prediction["id"].item()] |
|
output["nodes"] = self.create_nodes(prediction) |
|
output["nodes"] = self.create_anchors(prediction, output["nodes"], join_contiguous=True, at_least_one=True, mode="anchors") |
|
output["nodes"] = self.create_anchors(prediction, output["nodes"], join_contiguous=True, at_least_one=False, mode="source anchors") |
|
output["nodes"] = self.create_anchors(prediction, output["nodes"], join_contiguous=True, at_least_one=False, mode="target anchors") |
|
output["edges"], output["nodes"] = self.create_targets_sources(output["nodes"]) |
|
|
|
return output |
|
|
|
def create_targets_sources(self, nodes): |
|
edges, new_nodes = [], [] |
|
for i, node in enumerate(nodes): |
|
new_node_id = len(nodes) + len(new_nodes) |
|
if len(node["source anchors"]) > 0: |
|
new_nodes.append({"id": new_node_id, "label": "Source", "anchors": node["source anchors"]}) |
|
edges.append({"source": i, "target": new_node_id, "label": ""}) |
|
new_node_id += 1 |
|
del node["source anchors"] |
|
|
|
if len(node["target anchors"]) > 0: |
|
new_nodes.append({"id": new_node_id, "label": "Target", "anchors": node["target anchors"]}) |
|
edges.append({"source": i, "target": new_node_id, "label": ""}) |
|
del node["target anchors"] |
|
|
|
return edges, nodes + new_nodes |
|
|
