Package com.redis.vl.extensions.summarization

Class EmbeddedSentence

java.lang.Object

com.redis.vl.extensions.summarization.EmbeddedSentence

All Implemented Interfaces:

org.apache.commons.math3.ml.clustering.Clusterable

public class EmbeddedSentence
extends Object
implements org.apache.commons.math3.ml.clustering.Clusterable

A sentence with its embedding, implementing Clusterable for k-means.

Constructor Summary

Constructors

Constructor	Description
EmbeddedSentence(int index, float[] embedding)	Create an embedded sentence.

Method Summary

Modifier and Type	Method	Description
double	cosineSimilarity(EmbeddedSentence other)	Calculate cosine similarity with another embedded sentence.
double[]	getPoint()	Get the embedding as double array (required by Clusterable).
int	index()	Get the original index of this sentence.

Methods inherited from class java.lang.Object

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

Constructor Details

EmbeddedSentence

public EmbeddedSentence(int index,
 float[] embedding)

Create an embedded sentence.

Parameters:

index - Original index in the sentence list (for preserving order)

embedding - The BERT embedding as float array

Method Details

index

public int index()

Get the original index of this sentence.

getPoint

public double[] getPoint()

Get the embedding as double array (required by Clusterable).

Specified by:

getPoint in interface org.apache.commons.math3.ml.clustering.Clusterable

cosineSimilarity

public double cosineSimilarity(EmbeddedSentence other)

Calculate cosine similarity with another embedded sentence.