Gagan Bhatia commited on
Commit
43c82bc
1 Parent(s): 8484add

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
@@ -11,6 +11,7 @@ def make_dataset(dataset="cnn_dailymail", split="train"):
11
  os.makedirs("data/raw")
12
  dataset = load_dataset(dataset, "3.0.0", split=split)
13
  df = pd.DataFrame()
 
14
 
15
 
16
  if __name__ == '__main__':
 
11
  os.makedirs("data/raw")
12
  dataset = load_dataset(dataset, "3.0.0", split=split)
13
  df = pd.DataFrame()
14
+ df.to_csv("data/raw/{}.csv".format(split))
15
 
16
 
17
  if __name__ == '__main__':