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)