org.noear.solon.ai.rag.util

Class SimilarityMath

java.lang.Object

org.noear.solon.ai.rag.util.SimilarityMath

public final class SimilarityMath
extends Object

相似度算法

Constructor Summary

Constructors

Constructor and Description
SimilarityMath()

Method Summary

Modifier and Type	Method and Description
static double	cosineSimilarity(float[] embedA, float[] embedB) 余弦相似度
static float	dotProduct(float[] embedA, float[] embedB) 点积
static float	norm(float[] vector) 范数

Methods inherited from class java.lang.Object

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

Constructor Detail

SimilarityMath

public SimilarityMath()

Method Detail

cosineSimilarity

public static double cosineSimilarity(float[] embedA,
                                      float[] embedB)

余弦相似度

Parameters:

embedA - 嵌入矢量1

embedB - 嵌入矢量2

dotProduct

public static float dotProduct(float[] embedA,
                               float[] embedB)

点积

norm

public static float norm(float[] vector)

范数