Gagan Bhatia commited on
Commit
93aba5e
1 Parent(s): a06b193

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
@@ -5,6 +5,7 @@ import os
5
  import pprint
6
 
7
 
 
8
  """make dataset for summarisation"""
9
  if not os.path.exists('data/raw'):
10
  os.makedirs('data/raw')
 
5
  import pprint
6
 
7
 
8
+ def make_dataset(dataset="cnn_dailymail", split="train"):
9
  """make dataset for summarisation"""
10
  if not os.path.exists('data/raw'):
11
  os.makedirs('data/raw')