def load_rocstories_dataset(dataset_path):
    """Output a list of tuples(story, 1st continuation, 2nd continuation, label)"""
    with open(dataset_path, encoding="utf_8") as f:
        f = csv.reader(f)
        output = []
        next(f)  # skip the first line
        for line in tqdm(f):
            output.append((" ".join(line[1:5]), line[5], line[6], int(line[-1]) - 1))
    return output
Output a list of tuples(story, 1st continuation, 2nd continuation, label)
dpkd/transformers/examples/legacy/run_openai_gpt.py