Module: tfr.extension.premade.tfrbert_task

TF-Ranking BERT task.

Classes

class TFRBertConfig: The tf-ranking BERT task config.

class TFRBertDataConfig: Data config for TFR-BERT task.

class TFRBertDataLoader: A class to load dataset for TFR-BERT task.

class TFRBertModelBuilder: Model builder for TFR-BERT models.

class TFRBertModelConfig: A TFR-BERT model configuration.

class TFRBertScorer: Univariate BERT-based scorer.

class TFRBertTask: Task object for tf-ranking BERT.

Type Aliases

TensorDict

TensorLike

DOCUMENT_ID 'document_id'
QUERY_ID 'query_id'