java.lang.Object
org.elasticsearch.script.field.vectors.KnnDenseVector
- All Implemented Interfaces:
DenseVector
-
Field Summary
FieldsFields inherited from interface org.elasticsearch.script.field.vectors.DenseVector
EMPTY -
Constructor Summary
ConstructorsConstructorDescriptionKnnDenseVector(float[] docVector) KnnDenseVector(float[] docVector, float magnitude) -
Method Summary
Modifier and TypeMethodDescriptiondoublecosineSimilarity(byte[] queryVector, float qvMagnitude) Get the cosine similarity with the query vectordoublecosineSimilarity(float[] queryVector, boolean normalizeQueryVector) Get the cosine similarity with the query vectordoublecosineSimilarity(List<Number> queryVector) Get the cosine similarity with the un-normalized query vectorintdotProduct(byte[] queryVector) doubledotProduct(float[] queryVector) doubledotProduct(List<Number> queryVector) intgetDims()floatfloat[]inthamming(byte[] queryVector) intbooleanisEmpty()intl1Norm(byte[] queryVector) doublel1Norm(float[] queryVector) doubledoublel2Norm(byte[] queryVector) doublel2Norm(float[] queryVector) doubleintsize()Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, waitMethods inherited from interface org.elasticsearch.script.field.vectors.DenseVector
checkDimensions, cosineSimilarity, cosineSimilarity, cosineSimilarity, dotProduct, hamming, l1Norm, l2Norm
-
Field Details
-
docVector
protected final float[] docVector
-
-
Constructor Details
-
KnnDenseVector
public KnnDenseVector(float[] docVector) -
KnnDenseVector
public KnnDenseVector(float[] docVector, float magnitude)
-
-
Method Details
-
getVector
public float[] getVector()- Specified by:
getVectorin interfaceDenseVector
-
getMagnitude
public float getMagnitude()- Specified by:
getMagnitudein interfaceDenseVector
-
dotProduct
public int dotProduct(byte[] queryVector) - Specified by:
dotProductin interfaceDenseVector
-
dotProduct
public double dotProduct(float[] queryVector) - Specified by:
dotProductin interfaceDenseVector
-
dotProduct
- Specified by:
dotProductin interfaceDenseVector
-
l1Norm
public int l1Norm(byte[] queryVector) - Specified by:
l1Normin interfaceDenseVector
-
l1Norm
public double l1Norm(float[] queryVector) - Specified by:
l1Normin interfaceDenseVector
-
l1Norm
- Specified by:
l1Normin interfaceDenseVector
-
hamming
public int hamming(byte[] queryVector) - Specified by:
hammingin interfaceDenseVector
-
hamming
- Specified by:
hammingin interfaceDenseVector
-
l2Norm
public double l2Norm(byte[] queryVector) - Specified by:
l2Normin interfaceDenseVector
-
l2Norm
public double l2Norm(float[] queryVector) - Specified by:
l2Normin interfaceDenseVector
-
l2Norm
- Specified by:
l2Normin interfaceDenseVector
-
cosineSimilarity
public double cosineSimilarity(byte[] queryVector, float qvMagnitude) Description copied from interface:DenseVectorGet the cosine similarity with the query vector- Specified by:
cosineSimilarityin interfaceDenseVector- Parameters:
qvMagnitude- - pre-calculated magnitude of the query vector
-
cosineSimilarity
public double cosineSimilarity(float[] queryVector, boolean normalizeQueryVector) Description copied from interface:DenseVectorGet the cosine similarity with the query vector- Specified by:
cosineSimilarityin interfaceDenseVector- Parameters:
normalizeQueryVector- - normalize the query vector, does not change the contents of passed in query vector
-
cosineSimilarity
Description copied from interface:DenseVectorGet the cosine similarity with the un-normalized query vector- Specified by:
cosineSimilarityin interfaceDenseVector
-
isEmpty
public boolean isEmpty()- Specified by:
isEmptyin interfaceDenseVector
-
getDims
public int getDims()- Specified by:
getDimsin interfaceDenseVector
-
size
public int size()- Specified by:
sizein interfaceDenseVector
-