tfr.extension.premade.TFRBertDataLoader

Stay organized with collections Save and categorize content based on your preferences.

A class to load dataset for TFR-BERT task.

Inherits From: RankingDataLoader

Methods

load

View source

Returns a tf.dataset.Dataset.