Module org.apache.lucene.codecs

Package org.apache.lucene.codecs.bitvectors

Class HnswBitVectorsFormat

java.lang.Object

org.apache.lucene.codecs.KnnVectorsFormat

org.apache.lucene.codecs.bitvectors.HnswBitVectorsFormat

All Implemented Interfaces:

NamedSPILoader.NamedSPI

public final class HnswBitVectorsFormat
extends KnnVectorsFormat

Encodes bit vector values into an associated graph connecting the documents having values. The graph is used to power HNSW search. The format consists of two files, and uses Lucene99FlatVectorsFormat to store the actual vectors, but with a custom scorer implementation: For details on graph storage and file extensions, see Lucene99HnswVectorsFormat.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
private static class	HnswBitVectorsFormat.FlatBitVectorsWriter

Field Summary

Fields

Modifier and Type	Field	Description
private final int	beamWidth	The number of candidate neighbors to track while searching the graph for each newly inserted node.
private final FlatVectorsFormat	flatVectorsFormat	The format for storing, reading, merging vectors on disk
private final int	maxConn	Controls how many of the nearest neighbor candidates are connected to the new node.
private final TaskExecutor	mergeExec
static final String	NAME
private final int	numMergeWorkers

Fields inherited from class org.apache.lucene.codecs.KnnVectorsFormat

DEFAULT_MAX_DIMENSIONS, EMPTY

Constructor Summary

Constructors

Constructor	Description
HnswBitVectorsFormat()	Constructs a format using default graph construction parameters
HnswBitVectorsFormat(int maxConn, int beamWidth)	Constructs a format using the given graph construction parameters.
HnswBitVectorsFormat(int maxConn, int beamWidth, int numMergeWorkers, ExecutorService mergeExec)	Constructs a format using the given graph construction parameters and scalar quantization.

Method Summary

Modifier and Type	Method	Description
KnnVectorsReader	fieldsReader(SegmentReadState state)	Returns a KnnVectorsReader to read the vectors from the index.
KnnVectorsWriter	fieldsWriter(SegmentWriteState state)	Returns a KnnVectorsWriter to write the vectors to the index.
int	getMaxDimensions(String fieldName)	Returns the maximum number of vector dimensions supported by this codec for the given field name
String	toString()

Methods inherited from class org.apache.lucene.codecs.KnnVectorsFormat

availableKnnVectorsFormats, forName, getName, reloadKnnVectorsFormat

Methods inherited from class java.lang.Object

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

Field Details

NAME

public static final String NAME

See Also:

• Constant Field Values

maxConn

private final int maxConn

Controls how many of the nearest neighbor candidates are connected to the new node. Defaults to Lucene99HnswVectorsFormat.DEFAULT_MAX_CONN. See HnswGraph for more details.

beamWidth

private final int beamWidth

The number of candidate neighbors to track while searching the graph for each newly inserted node. Defaults to Lucene99HnswVectorsFormat.DEFAULT_BEAM_WIDTH. See HnswGraph for details.

flatVectorsFormat

private final FlatVectorsFormat flatVectorsFormat

The format for storing, reading, merging vectors on disk

numMergeWorkers

private final int numMergeWorkers

mergeExec

private final TaskExecutor mergeExec

Constructor Details

HnswBitVectorsFormat

public HnswBitVectorsFormat()

Constructs a format using default graph construction parameters

HnswBitVectorsFormat

public HnswBitVectorsFormat(int maxConn,
 int beamWidth)

Constructs a format using the given graph construction parameters.

Parameters:

maxConn - the maximum number of connections to a node in the HNSW graph

beamWidth - the size of the queue maintained during graph construction.

HnswBitVectorsFormat

public HnswBitVectorsFormat(int maxConn,
 int beamWidth,
 int numMergeWorkers,
 ExecutorService mergeExec)

Constructs a format using the given graph construction parameters and scalar quantization.

Parameters:

maxConn - the maximum number of connections to a node in the HNSW graph

beamWidth - the size of the queue maintained during graph construction.

numMergeWorkers - number of workers (threads) that will be used when doing merge. If larger than 1, a non-null ExecutorService must be passed as mergeExec

mergeExec - the ExecutorService that will be used by ALL vector writers that are generated by this format to do the merge

Method Details

fieldsWriter

public KnnVectorsWriter fieldsWriter(SegmentWriteState state)
                              throws IOException

Description copied from class: KnnVectorsFormat

Returns a KnnVectorsWriter to write the vectors to the index.

Specified by:

fieldsWriter in class KnnVectorsFormat

Throws:

IOException

fieldsReader

public KnnVectorsReader fieldsReader(SegmentReadState state)
                              throws IOException

Description copied from class: KnnVectorsFormat

Returns a KnnVectorsReader to read the vectors from the index.

Specified by:

fieldsReader in class KnnVectorsFormat

Throws:

IOException

getMaxDimensions

public int getMaxDimensions(String fieldName)

Description copied from class: KnnVectorsFormat

Returns the maximum number of vector dimensions supported by this codec for the given field name

Codecs should override this method to specify the maximum number of dimensions they support.

Overrides:

getMaxDimensions in class KnnVectorsFormat

Parameters:

fieldName - the field name

Returns:

the maximum number of vector dimensions.

toString

public String toString()

Overrides:

toString in class Object