Google I/O is a wrap! Catch up on TensorFlow sessions View sessions

tfr.extension.premade.TFRBertDataLoader

A class to load dataset for TFR-BERT task.

Inherits From: RankingDataLoader

Methods

load

View source

Returns a tf.dataset.Dataset.