tfr.extension.premade.TFRBertDataLoader

View source on GitHub

A class to load dataset for TFR-BERT task.

Inherits From: RankingDataLoader

tfr.extension.premade.TFRBertDataLoader(
    params, label_spec: Tuple[str, tf.io.FixedLenFeature] = None, **kwargs
)

Methods

load

View source

load(
    input_context: Optional[tf.distribute.InputContext] = None
) -> tf.data.Dataset

Returns a tf.dataset.Dataset.