parser
module wittgenstein_rebel_extractor.parser
Classes
-
KGTriplet — Pydantic model representing a Knowledge Graph triplet extracted from text.
Functions
-
parse_rebel_triplets — Parse raw decoded sequence output from the REBEL model into structured KGTriplet models.
wittgenstein_rebel_extractor.parser.KGTriplet
class KGTriplet()
Bases : BaseModel
Pydantic model representing a Knowledge Graph triplet extracted from text.
wittgenstein_rebel_extractor.parser.parse_rebel_triplets
parse_rebel_triplets(decoded_text: str) → List[KGTriplet]
Parse raw decoded sequence output from the REBEL model into structured KGTriplet models.
The expected format is