RelikGraphTransformer#
- class langchain_experimental.graph_transformers.relik.RelikGraphTransformer(model: str = 'relik-ie/relik-relation-extraction-small', relationship_confidence_threshold: float = 0.1, model_config: Dict[str, Any] = {}, ignore_self_loops: bool = True)[source]#
一个用于将文档转换为图结构的转换器类，使用Relik库和模型。
此类利用relik模型从文本文档中提取关系和节点，并将它们转换为图形格式。关系根据指定的置信度阈值进行过滤。
- For more details on the Relik library, visit their GitHub repository:
- Parameters:
model (str) – 使用的预训练Relik模型的名称。 默认是“relik-ie/relik-relation-extraction-small-wikipedia”。
relationship_confidence_threshold (float) – 用于过滤关系的置信度阈值。默认值为0.1。
model_config (Dict[str, any]) – Relik 模型的额外配置选项。默认是一个空字典。
ignore_self_loops (bool) – 是否忽略源节点和目标节点相同的关系。默认值为 True。
方法
__init__([model, ...])convert_to_graph_documents(documents)将一系列文档转换为图形文档。
process_document(document)- __init__(model: str = 'relik-ie/relik-relation-extraction-small', relationship_confidence_threshold: float = 0.1, model_config: Dict[str, Any] = {}, ignore_self_loops: bool = True) None[来源]#
- Parameters:
model (str)
relationship_confidence_threshold (float)
model_config (Dict[str, Any])
ignore_self_loops (bool)
- Return type:
无
- convert_to_graph_documents(documents: Sequence[Document]) List[GraphDocument][source]#
将一系列文档转换为图形文档。
- Parameters:
documents (Sequence[Document]) – 原始文档。
kwargs – 额外的关键字参数。
- Returns:
将文档转换为图形。
- Return type:
序列[GraphDocument]
- process_document(document: Document) GraphDocument[source]#
- Parameters:
文档 (Document)
- Return type: