forked from thuml/Time-Series-Library
-
Notifications
You must be signed in to change notification settings - Fork 1
/
contact_data.py
36 lines (30 loc) · 1.31 KB
/
contact_data.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
25
26
27
28
29
30
31
32
33
34
35
36
import os
from optimizers.optimizer_transformer import h
def contact_data_to_file(core_process_file, other_process_files):
print(other_process_files + ' to ' + core_process_file)
if os.path.exists(core_process_file):
if os.path.exists(other_process_files):
with open(other_process_files, "r") as f:
with open(core_process_file, "a") as f1:
# skip the first line and write the remaining lines
f.readline()
for line in f:
f1.write(line)
# delete file
os.remove(other_process_files)
else:
if os.path.exists(other_process_files):
with open(other_process_files, "r") as f:
with open(core_process_file, "a") as f1:
for line in f:
f1.write(line)
# delete file
os.remove(other_process_files)
optimizer_settings = h.get_optimizer_settings()
task_names = h.get_all_task_names()
_max_process_index = optimizer_settings['process_number'] - 1
for task_name in task_names:
for process_index in range(_max_process_index):
if process_index == 0:
continue
contact_data_to_file(h.get_csv_file_path(task_name), h.get_csv_file_path(task_name, process_index))