Module: tfr.extension.task

Orbit task for TF-Ranking.

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

DATASET_FN_MAP

{
 'textline': <class 'tensorflow.python.data.ops.readers.TextLineDatasetV2'>,
 'tfrecord': <class 'tensorflow.python.data.ops.readers.TFRecordDatasetV2'>
}

MASK 'example_list_mask'