java.lang.Object
org.apache.lucene.index.KnnVectorValues
org.apache.lucene.index.FloatVectorValues
org.elasticsearch.index.codec.vectors.SampleReader
- All Implemented Interfaces:
org.apache.lucene.codecs.lucene95.HasIndexSlice
public class SampleReader
extends org.apache.lucene.index.FloatVectorValues
implements org.apache.lucene.codecs.lucene95.HasIndexSlice
-
Nested Class Summary
Nested classes/interfaces inherited from class org.apache.lucene.index.KnnVectorValues
org.apache.lucene.index.KnnVectorValues.DocIndexIterator -
Method Summary
Modifier and TypeMethodDescriptionorg.apache.lucene.index.FloatVectorValuescopy()static SampleReadercreateSampleReader(org.apache.lucene.index.FloatVectorValues origin, int k, long seed) intorg.apache.lucene.util.BitsgetAcceptOrds(org.apache.lucene.util.Bits acceptDocs) org.apache.lucene.store.IndexInputgetSlice()intintordToDoc(int ord) static int[]reservoirSample(int n, int k, long seed) Sample k elements from n elements according to reservoir sampling algorithm.intsize()float[]vectorValue(int targetOrd) Methods inherited from class org.apache.lucene.index.FloatVectorValues
checkField, fromFloats, getEncoding, scorerMethods inherited from class org.apache.lucene.index.KnnVectorValues
createDenseIterator, createSparseIterator, fromDISI, iterator
-
Method Details
-
size
public int size()- Specified by:
sizein classorg.apache.lucene.index.KnnVectorValues
-
dimension
public int dimension()- Specified by:
dimensionin classorg.apache.lucene.index.KnnVectorValues
-
copy
- Specified by:
copyin classorg.apache.lucene.index.FloatVectorValues- Throws:
IOException
-
getSlice
public org.apache.lucene.store.IndexInput getSlice()- Specified by:
getSlicein interfaceorg.apache.lucene.codecs.lucene95.HasIndexSlice
-
vectorValue
- Specified by:
vectorValuein classorg.apache.lucene.index.FloatVectorValues- Throws:
IOException
-
getVectorByteLength
public int getVectorByteLength()- Overrides:
getVectorByteLengthin classorg.apache.lucene.index.KnnVectorValues
-
ordToDoc
public int ordToDoc(int ord) - Overrides:
ordToDocin classorg.apache.lucene.index.KnnVectorValues
-
getAcceptOrds
public org.apache.lucene.util.Bits getAcceptOrds(org.apache.lucene.util.Bits acceptDocs) - Overrides:
getAcceptOrdsin classorg.apache.lucene.index.KnnVectorValues
-
createSampleReader
public static SampleReader createSampleReader(org.apache.lucene.index.FloatVectorValues origin, int k, long seed) -
reservoirSample
public static int[] reservoirSample(int n, int k, long seed) Sample k elements from n elements according to reservoir sampling algorithm.- Parameters:
n- number of elementsk- number of samplesseed- random seed- Returns:
- array of k samples
-