vertebrae.compression.base

Embedding compression adapters and utilities.

Classes

CompressionResult

Compressed embedding matrix plus serializable metadata.

EmbeddingCompressor

Protocol-style base for embedding compressors.

Functions

compress_embeddings(Z[, config, y])

Compress embeddings according to a compression config.

Module Contents

class vertebrae.compression.base.CompressionResult[source]

Compressed embedding matrix plus serializable metadata.

class vertebrae.compression.base.EmbeddingCompressor[source]

Protocol-style base for embedding compressors.

vertebrae.compression.base.compress_embeddings(Z, config=None, y=None)[source]

Compress embeddings according to a compression config.

Parameters:
Return type:

CompressionResult