com.huaweicloud.pangu.dev.sdk.retriever

Class KGRetriever

java.lang.Object

com.huaweicloud.pangu.dev.sdk.retriever.KGRetriever

All Implemented Interfaces:

Retriever

public class KGRetriever
extends Object
implements Retriever

知识图谱检索工具

Since:

2023-11-07

Constructor Summary

Constructors

Constructor and Description
KGRetriever(KnowledgeGraph kg) 初始化

Method Summary

Modifier and Type	Method and Description
List<Document>	getRelevantDocument(String query) 检索相关文档
List<Document>	getRelevantDocument(String query, int topK) 检索相关文档(topK条)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

KGRetriever

public KGRetriever(KnowledgeGraph kg)

初始化

Parameters:

kg - 知识图谱引擎

Method Detail

getRelevantDocument

public List<Document> getRelevantDocument(String query)

Description copied from interface: Retriever

检索相关文档

Specified by:

getRelevantDocument in interface Retriever

Parameters:

query - 检索文本

Returns:

相关性文档

getRelevantDocument

public List<Document> getRelevantDocument(String query,
                                          int topK)

Description copied from interface: Retriever

检索相关文档(topK条)

Specified by:

getRelevantDocument in interface Retriever

Parameters:

query - 检索文本

topK - topK条记录

Returns:

相关性文档