<gh_stars>0
train_file = open('train.txt', 'w')
val_file = open('val.txt', 'w')
from sklearn.model_selection import train_test_split
file_ids = ["%06d\n" % i for i in range(0,7481)]
train_ids, val_ids = train_test_split(file_ids, test_size=0.5037)
print(train_ids)
train_file.writelines(train_ids)
val_file.writelines(val_ids)
train_file.close()
val_file.close()
