File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -117,12 +117,13 @@ def relational_pipeline(
117117 :param ner_kwargs: kwargs for :class:`.BertNamedEntityRecognizer`
118118 :param characters_unifier_kwargs: kwargs for
119119 :class:`.GraphRulesCharacterUnifier`
120- :param relation_extraction_kwargs: kwargs for :class:`.T5RelationExtractor`
120+ :param relation_extraction_kwargs: kwargs for
121+ :class:`.GenerativeRelationExtractor`
121122 :param pipeline_kwargs: kwargs for :class:`.Pipeline`
122123 """
123124 from renard .pipeline .tokenization import NLTKTokenizer
124125 from renard .pipeline .character_unification import GraphRulesCharacterUnifier
125- from renard .pipeline .relation_extraction import T5RelationExtractor
126+ from renard .pipeline .relation_extraction import GenerativeRelationExtractor
126127 from renard .pipeline .graph_extraction import RelationalGraphExtractor
127128
128129 tokenizer_kwargs = tokenizer_kwargs or {}
@@ -136,7 +137,7 @@ def relational_pipeline(
136137 NLTKTokenizer (** tokenizer_kwargs ),
137138 ner_step (** ner_kwargs ),
138139 GraphRulesCharacterUnifier (** character_unifier_kwargs ),
139- T5RelationExtractor (** relation_extractor_kwargs ),
140+ GenerativeRelationExtractor (** relation_extractor_kwargs ),
140141 RelationalGraphExtractor (** graph_extractor_kwargs ),
141142 ],
142143 ** pipeline_kwargs ,
You can’t perform that action at this time.
0 commit comments