Module: tfr.extension.task

Orbit task for TF-Ranking.

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.

Classes

class RankingDataConfig: Data set config.

class RankingDataLoader: A class to load dataset for ranking task.

class RankingTask: Task object for TF-Ranking.

class RankingTaskConfig: The TF-Ranking task config.

Type Aliases

FeatureSpec

Other Members
DATASET_FN_MAP	{ 'textline': <class 'tensorflow.python.data.ops.readers.TextLineDatasetV2'>, 'tfrecord': <class 'tensorflow.python.data.ops.readers.TFRecordDatasetV2'> }
MASK	'example_list_mask'