Skip to content

Commit fe7f47d

Browse files
Julian Kates-HarbeckJulian Kates-Harbeck
authored andcommitted
adding bleed in to train and validation sets
1 parent 7cfbf53 commit fe7f47d

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

plasma/preprocessor/preprocess.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,7 @@ def apply_bleed_in(conf,shot_list_train,shot_list_validate,shot_list_test):
153153
for i in range(num):
154154
s = shot_list_test.sample_single_class(True)
155155
shot_list_train.append(s)
156+
shot_list_validate.append(s)
156157
if conf['data']['bleed_in_remove_from_test']:
157158
shot_list_test.remove(s)
158159
else:
@@ -161,6 +162,7 @@ def apply_bleed_in(conf,shot_list_train,shot_list_validate,shot_list_test):
161162
for i in range(num):
162163
s = shot_list_test.sample_single_class(False)
163164
shot_list_train.append(s)
165+
shot_list_validate.append(s)
164166
if conf['data']['bleed_in_remove_from_test']:
165167
shot_list_test.remove(s)
166168
else:

0 commit comments

Comments
 (0)