-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathrun_preprocessing.py
24 lines (18 loc) · 1.1 KB
/
run_preprocessing.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
from own_package.preprocessing import read_excel_data_to_cutoff, read_grid_data, l2_tracker
from own_package.others import create_results_directory
def run_preprocess(select):
# Selector to choose which code from preprocessing to run.
# Here, we only change the file directory to choose which excel file is inputted into the various functions
if select == 1:
write_dir = create_results_directory('./results/preprocessing/preprocess_round13')
read_excel_data_to_cutoff(read_excel_file='./excel/Raw_Data_Round13.xlsx',
write_dir=write_dir)
elif select == 2:
write_dir = create_results_directory('./results/preprocessing/grid')
read_grid_data(read_excel_file='./excel/grid.xlsx', write_dir=write_dir)
elif select == 3:
l2_tracker(write_excel_dir='./results/preprocessing/l2_information_rounded.xlsx',
final_excel_loader='./excel/Data_loader_Round13_rounded.xlsx',
last_idx_store=[11, 16, 21, 29, 37, 45, 69, 77, 85, 93, 101, 109, 117, 125])
if __name__ == '__main__':
run_preprocess(3)