set_encoder

Configuration and model implementation for SetEncoder type models. Originally proposed in Set-Encoder: Permutation-Invariant Inter-passage Attention for Listwise Passage Re-ranking with Cross-Encoders.

Classes

SetEncoderConfig(*args[, depth, ...])

Configuration class for a SetEncoder model.

SetEncoderModel(config, *args, **kwargs)

SetEncoder model.

SetEncoderTokenizer(*args[, query_length, ...])