Module: tfr.extension.premade

TensorFlow Ranking Premade Orbit Task Module.

Note: First - These APIs require These APS require the tensorflow_modelspackage. You can install it with pip install tf-models-official. Second - Nothing under tensorflow_ranking.extension.premade is imported by default. To use these APIs import premade in your code: import tensorflow_ranking.extension.premade.

Modules

tfrbert_task module: 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

Other Members
DOCUMENT_ID	'document_id'
QUERY_ID	'query_id'