SeismicFormatConverter

class lightning_ir.retrieve.seismic.seismic_format.SeismicFormatConverter

Bases: object

Converter for embeddings to a seismic format.

Methods

convert_to_seismic_format(embeddings)

Convert embeddings to a seismic format.

static convert_to_seismic_format(embeddings: Tensor) → bytes

Convert embeddings to a seismic format.

Parameters:

embeddings (torch.Tensor) – The embeddings to convert.

Returns:

The converted embeddings in seismic format.

Return type:

bytes

Raises:

ValueError – If the embeddings are not 2D.