Gagan Bhatia commited on
Commit
eb71d14
1 Parent(s): 8cab515

Update make_dataset.py

Browse files
Files changed (1) hide show
  1. src/data/make_dataset.py +1 -0
src/data/make_dataset.py CHANGED
@@ -7,3 +7,4 @@ def make_dataset(dataset='cnn_dailymail', split='train', version="3.0.0"):
7
  dataset = load_dataset(dataset, split=split, script_version=version)
8
  df = pd.DataFrame()
9
  df['input_text'] = dataset['concepts']
 
 
7
  dataset = load_dataset(dataset, split=split, script_version=version)
8
  df = pd.DataFrame()
9
  df['input_text'] = dataset['concepts']
10
+ df['output_text'] = dataset['target']