model

Model module for cross-encoder models.

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

Classes

CrossEncoderModel(config, *args, **kwargs)

CrossEncoderOutput([scores, embeddings])

Dataclass containing the output of a cross-encoder model