def make_dataset(self, data_file, training=None):
if not isinstance(data_file, list):
return super().make_dataset(data_file, training=training)
text_dataset = super().make_dataset(data_file[0], training=training)
if training:
return text_dataset
ref_dataset = make_datasets(tf.data.TextLineDataset, data_file[1])
return tf.data.Dataset.zip((text_dataset, ref_dataset))
def make_dataset(self, data_file, training=None):
return dataset_util.make_datasets(tf.data.TextLineDataset, data_file)
def make_dataset(self, data_file, training=None):
return dataset_util.make_datasets(tf.data.TFRecordDataset, data_file)
