gagan3012 commited on
Commit
5322cc9
·
1 Parent(s): 6d7cb20

Update process_data.py

Browse files
Files changed (1) hide show
  1. src/data/process_data.py +2 -0
src/data/process_data.py CHANGED
@@ -14,3 +14,5 @@ if __name__ == "__main__":
14
  params = yaml.safe_load(f)
15
 
16
  process_data(frac=params["split"], split="train")
 
 
 
14
  params = yaml.safe_load(f)
15
 
16
  process_data(frac=params["split"], split="train")
17
+ process_data(frac=params["split"], split="test")
18
+ process_data(frac=params["split"], split="validation")