Gagan Bhatia commited on
Commit
483a16d
1 Parent(s): 676b027

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
@@ -8,6 +8,7 @@ import pprint
8
  def make_dataset(dataset="cnn_dailymail", split="train"):
9
  """make dataset for summarisation"""
10
  if not os.path.exists("data/raw"):
 
11
  df = pd.DataFrame()
12
  df['article'] = dataset['article']
13
  df['highlights'] = dataset['highlights']
 
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")
12
  df = pd.DataFrame()
13
  df['article'] = dataset['article']
14
  df['highlights'] = dataset['highlights']