Andrea Maldonado commited on
Commit
6e9d02e
·
1 Parent(s): 9b61a3f

Passes arg to grid experiments script

Browse files
.github/workflows/test_gedi.yml CHANGED
@@ -194,7 +194,7 @@ jobs:
194
 
195
  - name: Run test
196
  run:
197
- python execute_grid_experiments.py
198
 
199
  - name: Convert output and validation to same encoding
200
  run: iconv -f UTF-8 -t ASCII output/features/generated/2_bpic_features/2_ense_enseef_feat.csv > data/validation/2_ense_enseef_feat.csv
 
194
 
195
  - name: Run test
196
  run:
197
+ python execute_grid_experiments.py config_files/algorithm/test
198
 
199
  - name: Convert output and validation to same encoding
200
  run: iconv -f UTF-8 -t ASCII output/features/generated/2_bpic_features/2_ense_enseef_feat.csv > data/validation/2_ense_enseef_feat.csv
execute_grid_experiments.py CHANGED
@@ -1,5 +1,6 @@
1
  import multiprocessing
2
  import os
 
3
 
4
  from datetime import datetime as dt
5
  from gedi.utils.io_helpers import sort_files
@@ -13,10 +14,13 @@ def multi_experiment_wrapper(config_file, i=0):
13
  print(f"=========================FINISHED EXPERIMENT #{i+1}=======================")
14
 
15
  if __name__ == '__main__':
16
- EXPERIMENTS_FOLDER = os.path.join('config_files','algorithm','BaselineED_feat')
17
- EXPERIMENTS_FOLDER = os.path.join('config_files','algorithm','grid_1obj')
18
- EXPERIMENTS_FOLDER = os.path.join('config_files','algorithm','grid_experiments')
19
- EXPERIMENTS_FOLDER = os.path.join('config_files','algorithm','test')
 
 
 
20
  start = dt.now()
21
 
22
  experiment_list = list(tqdm(sort_files(os.listdir(EXPERIMENTS_FOLDER))))
 
1
  import multiprocessing
2
  import os
3
+ import sys
4
 
5
  from datetime import datetime as dt
6
  from gedi.utils.io_helpers import sort_files
 
14
  print(f"=========================FINISHED EXPERIMENT #{i+1}=======================")
15
 
16
  if __name__ == '__main__':
17
+ EXPERIMENTS_FOLDER = sys.argv[1]
18
+ """
19
+ Following args run the following experiments:
20
+ - config_files/algorithm/grid_1obj
21
+ - config_files/algorithm/grid_experiments
22
+ - config_files/algorithm/test
23
+ """
24
  start = dt.now()
25
 
26
  experiment_list = list(tqdm(sort_files(os.listdir(EXPERIMENTS_FOLDER))))