constituency_parser
allennlp_models.structured_prediction.predictors.constituency_parser
LINK_TO_LABEL#
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]