gagan3012 commited on
Commit
a6ef8a4
·
1 Parent(s): b1cba3a
Files changed (2) hide show
  1. params.yml +1 -1
  2. src/data/make_dataset.py +1 -1
params.yml CHANGED
@@ -8,5 +8,5 @@ epochs: 5
8
  source_dir: src
9
  model_dir: models
10
  metric: rouge
11
- split: 0.001
12
  use_gpu: True
 
8
  source_dir: src
9
  model_dir: models
10
  metric: rouge
11
+ split: 0.01
12
  use_gpu: True
src/data/make_dataset.py CHANGED
@@ -19,7 +19,7 @@ def make_dataset(dataset='cnn_dailymail', split='train'):
19
  if __name__ == '__main__':
20
  with open("params.yml") as f:
21
  params = yaml.safe_load(f)
22
-
23
  make_dataset(dataset=params['data'], split='train')
24
  make_dataset(dataset=params['data'], split='test')
25
  make_dataset(dataset=params['data'], split='validation')
 
19
  if __name__ == '__main__':
20
  with open("params.yml") as f:
21
  params = yaml.safe_load(f)
22
+ print(params)
23
  make_dataset(dataset=params['data'], split='train')
24
  make_dataset(dataset=params['data'], split='test')
25
  make_dataset(dataset=params['data'], split='validation')