org.noear.solon.ai.rag.util

Class SimilarityUtil

java.lang.Object

org.noear.solon.ai.rag.util.SimilarityUtil

public final class SimilarityUtil
extends Object

相似度工具

Since:

3.1

Author:

noear

Constructor Summary

Constructors

Constructor and Description
SimilarityUtil()

Method Summary

Modifier and Type	Method and Description
static Document	copyAndScore(Document doc, float[] queryEmbed) 复制并评分
static List<Document>	filter(QueryCondition condition, Stream<Document> docs) 过滤（已经有评分的）
static List<Document>	scoreAndfilter(QueryCondition condition, Stream<Document> docs, float[] queryEmbed) 评分并过滤

Methods inherited from class java.lang.Object

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

Constructor Detail

SimilarityUtil

public SimilarityUtil()

Method Detail

filter

public static List<Document> filter(QueryCondition condition,
                                    Stream<Document> docs)
                             throws IOException

过滤（已经有评分的）

Throws:

IOException

scoreAndfilter

public static List<Document> scoreAndfilter(QueryCondition condition,
                                            Stream<Document> docs,
                                            float[] queryEmbed)
                                     throws IOException

评分并过滤

Throws:

IOException

copyAndScore

public static Document copyAndScore(Document doc,
                                    float[] queryEmbed)

复制并评分