-
Notifications
You must be signed in to change notification settings - Fork 0
/
split_train_data.py
33 lines (26 loc) · 869 Bytes
/
split_train_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
with open('/data/sqy/code/miccai2019/train_test_4/train_0.txt', 'r') as f:
lines = f.readlines()
pos_list = []
neg_list = []
for line in lines:
label = line.split('/')[-2]
if label == 'sig-train-neg':
neg_list.append(line)
else:
pos_list.append(line)
half_pos_num = int(len(pos_list) / 2)
half_neg_num = int(len(neg_list) / 2)
train1_list = pos_list[:half_pos_num]
train1_list.extend(neg_list[:half_neg_num])
train2_list = pos_list[half_pos_num:]
train2_list.extend(neg_list[half_neg_num:])
with open('/data/sqy/code/miccai2019/train_test_4/train_0_0.txt', 'w') as f:
input_str = ''
for i in train1_list:
input_str += i
f.write(input_str)
with open('/data/sqy/code/miccai2019/train_test_4/train_0_1.txt', 'w') as f:
input_str = ''
for i in train2_list:
input_str += i
f.write(input_str)