code: def load_rocstories_dataset(dataset_path):
      
      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
docs: Output a list of tuples(story, 1st continuation, 2nd continuation, label)