Gagan Bhatia commited on
Commit
a231173
1 Parent(s): 24a726e

Update make_dataset.py

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