Python DatasetSource.make_source_and_target Examples

Example #1

File: train.py Project: pursu/wavenet

    def eval_input_fn():
        eval_files_copy = list(eval_files)
        shuffle(eval_files_copy)

        eval_files_copy = tf.data.TFRecordDataset(eval_files_copy)

        dataset = DatasetSource(eval_files_copy, hparams)
        dataset = dataset.make_source_and_target().filter_by_max_output_length(
        ).repeat().group_by_batch(batch_size=1)
        return dataset.dataset

Example #2

File: predict.py Project: pursu/wavenet

 def predict_input_fn():
     records = tf.data.TFRecordDataset(list(test_files))
     dataset = DatasetSource(records, hparams)
     batched = dataset.make_source_and_target().group_by_batch(
         batch_size=1).arrange_for_prediction()
     return batched.dataset