Gagan Bhatia commited on
Commit
bf1265d
1 Parent(s): 08f7fc3

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
@@ -20,6 +20,7 @@ def make_dataset(dataset='cnn_dailymail', split='train'):
20
  if __name__ == '__main__':
21
  with open("params.yml") as f:
22
  params = yaml.safe_load(f)
 
23
  make_dataset(dataset=params['data'], split='train')
24
  make_dataset(dataset=params['data'], split='test')
25
  make_dataset(dataset=params['data'], split='validation')
 
20
  if __name__ == '__main__':
21
  with open("params.yml") as f:
22
  params = yaml.safe_load(f)
23
+ pprint.pprint(params)
24
  make_dataset(dataset=params['data'], split='train')
25
  make_dataset(dataset=params['data'], split='test')
26
  make_dataset(dataset=params['data'], split='validation')