Skip to content

constituency_parser

allennlp_models.structured_prediction.predictors.constituency_parser

[SOURCE]


LINK_TO_LABEL = {x: "pos" for x in TAG_MAP}

NODE_TYPE_TO_STYLE#

NODE_TYPE_TO_STYLE = {x: ["color0"] for x in TAG_MAP}

NODE_TYPE_TO_STYLE["NP"]#

NODE_TYPE_TO_STYLE["NP"] = ["color1"]

NODE_TYPE_TO_STYLE["NX"]#

NODE_TYPE_TO_STYLE["NX"] = ["color1"]

NODE_TYPE_TO_STYLE["QP"]#

NODE_TYPE_TO_STYLE["QP"] = ["color1"]

NODE_TYPE_TO_STYLE["NAC"]#

NODE_TYPE_TO_STYLE["NAC"] = ["color1"]

NODE_TYPE_TO_STYLE["VP"]#

NODE_TYPE_TO_STYLE["VP"] = ["color2"]

NODE_TYPE_TO_STYLE["S"]#

NODE_TYPE_TO_STYLE["S"] = ["color3"]

NODE_TYPE_TO_STYLE["SQ"]#

NODE_TYPE_TO_STYLE["SQ"] = ["color3"]

NODE_TYPE_TO_STYLE["SBAR"]#

NODE_TYPE_TO_STYLE["SBAR"] = ["color3"]

NODE_TYPE_TO_STYLE["SBARQ"]#

NODE_TYPE_TO_STYLE["SBARQ"] = ["color3"]

NODE_TYPE_TO_STYLE["SINQ"]#

NODE_TYPE_TO_STYLE["SINQ"] = ["color3"]

NODE_TYPE_TO_STYLE["FRAG"]#

NODE_TYPE_TO_STYLE["FRAG"] = ["color3"]

NODE_TYPE_TO_STYLE["X"]#

NODE_TYPE_TO_STYLE["X"] = ["color3"]

NODE_TYPE_TO_STYLE["WHADVP"]#

NODE_TYPE_TO_STYLE["WHADVP"] = ["color4"]

NODE_TYPE_TO_STYLE["WHADJP"]#

NODE_TYPE_TO_STYLE["WHADJP"] = ["color4"]

NODE_TYPE_TO_STYLE["WHNP"]#

NODE_TYPE_TO_STYLE["WHNP"] = ["color4"]

NODE_TYPE_TO_STYLE["WHPP"]#

NODE_TYPE_TO_STYLE["WHPP"] = ["color4"]

NODE_TYPE_TO_STYLE["PP"]#

NODE_TYPE_TO_STYLE["PP"] = ["color6"]

NODE_TYPE_TO_STYLE["ADJP"]#

NODE_TYPE_TO_STYLE["ADJP"] = ["color5"]

NODE_TYPE_TO_STYLE["ADVP"]#

NODE_TYPE_TO_STYLE["ADVP"] = ["color5"]

NODE_TYPE_TO_STYLE["CONJP"]#

NODE_TYPE_TO_STYLE["CONJP"] = ["color5"]

NODE_TYPE_TO_STYLE["INTJ"]#

NODE_TYPE_TO_STYLE["INTJ"] = ["color5"]

NODE_TYPE_TO_STYLE["LST"]#

NODE_TYPE_TO_STYLE["LST"] = ["color5", "seq"]

NODE_TYPE_TO_STYLE["PRN"]#

NODE_TYPE_TO_STYLE["PRN"] = ["color5"]

NODE_TYPE_TO_STYLE["PRT"]#

NODE_TYPE_TO_STYLE["PRT"] = ["color5"]

NODE_TYPE_TO_STYLE["RRC"]#

NODE_TYPE_TO_STYLE["RRC"] = ["color5"]

NODE_TYPE_TO_STYLE["UCP"]#

NODE_TYPE_TO_STYLE["UCP"] = ["color5"]

ConstituencyParserPredictor#

@Predictor.register("constituency_parser")
class ConstituencyParserPredictor(Predictor):
 | def __init__(
 |     self,
 |     model: Model,
 |     dataset_reader: DatasetReader,
 |     language: str = "en_core_web_sm"
 | ) -> None

Predictor for the SpanConstituencyParser model.

predict#

class ConstituencyParserPredictor(Predictor):
 | ...
 | def predict(self, sentence: str) -> JsonDict

Predict a constituency parse for the given sentence.

Parameters

  • sentence : str
    The sentence to parse.

Returns

  • A dictionary representation of the constituency tree.

predict_instance#

class ConstituencyParserPredictor(Predictor):
 | ...
 | def predict_instance(self, instance: Instance) -> JsonDict

predict_batch_instance#

class ConstituencyParserPredictor(Predictor):
 | ...
 | def predict_batch_instance(
 |     self,
 |     instances: List[Instance]
 | ) -> List[JsonDict]