model

Model module for bi-encoder models.

This module defines the model class used to implement bi-encoder models.

Classes

BiEncoderEmbedding(embeddings, scoring_mask, ...)	Dataclass containing embeddings and scoring mask for bi-encoder models.
BiEncoderModel(config, *args, **kwargs)
BiEncoderOutput([scores, query_embeddings, ...])	Dataclass containing the output of a bi-encoder model.
MLMHead(config)
ScoringFunction(config)