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>	refilter(Stream<Document> docs) 再过滤（评分与数量并排序）
static List<Document>	refilter(Stream<Document> docs, int limit) 再过滤（评分与数量并排序）
static List<Document>	refilter(Stream<Document> docs, int limit, double similarityThreshold) 再过滤（评分与数量并排序）
static List<Document>	refilter(Stream<Document> docs, QueryCondition condition) 再过滤
static Document	score(Document doc, float[] queryEmbed) 评分（评分与数量并排序）
static boolean	similarityCheck(Document doc, double similarityThreshold) 相似度检测

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

refilter

public static List<Document> refilter(Stream<Document> docs)

再过滤（评分与数量并排序）

refilter

public static List<Document> refilter(Stream<Document> docs,
                                      int limit)

再过滤（评分与数量并排序）

refilter

public static List<Document> refilter(Stream<Document> docs,
                                      int limit,
                                      double similarityThreshold)

再过滤（评分与数量并排序）

refilter

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

再过滤

Throws:

IOException

score

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

评分（评分与数量并排序）

copyAndScore

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

复制并评分

similarityCheck

public static boolean similarityCheck(Document doc,
                                      double similarityThreshold)

相似度检测