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.