vertebrae.compression.paired

Shared fitted compression for paired embedding endpoints.

Functions

compress_embedding_pair(fit_embeddings, ...[, ...])

Fit compression on one endpoint and apply it to a paired endpoint.

Module Contents

vertebrae.compression.paired.compress_embedding_pair(fit_embeddings, paired_embeddings, config, *, fit_name='fit embeddings', paired_name='paired embeddings')

Fit compression on one endpoint and apply it to a paired endpoint.

Parameters:

• fit_embeddings (Any)

• paired_embeddings (Any)

• config (vertebrae.config.EmbeddingCompressionConfig)

• fit_name (str)

• paired_name (str)

Return type:

Tuple[Any, Any, Dict[str, Any]]