Script to convert summarization data to tf_records.

create_input_data.py

import pandas as pd
import tensorflow as tf

save_path = "<Your path>/pegasus/data/testdata/test_pattern_1.tfrecords"

input_dict = dict(
                  inputs=[
                          # Your text inputs to be summarized.
                         ],
                  targets=[
                          # Corresponding targets for the inputs.
                          ]
                 )

data = pd.DataFrame(input_dict)

with tf.io.TFRecordWriter(save_path) as writer:
    for row in data.values:
        inputs, targets = row[:-1], row[-1]
        example = tf.train.Example(
            features=tf.train.Features(
                feature={
                    "inputs": tf.train.Feature(bytes_list=tf.train.BytesList(value=[inputs[0].encode('utf-8')])),
                    "targets": tf.train.Feature(bytes_list=tf.train.BytesList(value=[targets.encode('utf-8')])),
                }
            )
        )
        writer.write(example.SerializeToString())