coil

Configuration, model, and embedding for COIL (Contextualized Inverted List) type models. Originally proposed in COIL: Revisit Exact Lexical Match in Information Retrieval with Contextualized Inverted List.

Classes

CoilConfig([query_length, doc_length, ...])

Configuration class for COIL models.

CoilEmbedding(embeddings[, scoring_mask, ...])

Dataclass containing embeddings and the encoding for COIL models.

CoilModel(config, *args, **kwargs)

Multi-vector COIL model.

CoilOutput([scores, query_embeddings, ...])

Dataclass containing the output of a COIL model.

UniCoilConfig([query_length, doc_length, ...])

Configuration class for UniCOIL models.

UniCoilModel(config, *args, **kwargs)

Single-vector UniCOIL model.