langchain_experimental.graph_transformers.relik.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 模型从文本文档中提取关系和节点，并将其转换为图格式。根据指定的置信度阈值过滤关系。

有关 Relik 库的更多详情，请访问他们的 GitHub 仓库

https://github.com/SapienzaNLP/relik

参数

  • 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[source]
参数

  • model (str) –

  • relationship_confidence_threshold (float) –

  • model_config (Dict[str, Any]) –

  • ignore_self_loops (bool) –

返回类型

None

convert_to_graph_documents(documents: Sequence[Document]) → List[GraphDocument][source]

将一系列文档转换为图文档。

参数

  • documents (Sequence[Document]) – 原始文档。

  • kwargs – 其他关键字参数。

返回

转换为图文档的变换文档。

返回类型

Sequence[GraphDocument]

process_document(document: Document) → GraphDocument[source]
参数

document (Document) –

返回类型

GraphDocument