split train val dataset

import os
import random

import hashlib

f1 = open("train2.txt","w")
f2 = open("val2.txt","w")
lst = os.listdir("./")
random.shuffle(lst)
for l in lst:
    a = hashlib.md5(l.encode("utf-8"))
    b = a.hexdigest()
    as_int = int(b, 16)
    id = as_int%10
    if id<=8:
        f1.write(l+" "+str(as_int)+"\n")
    else:
        f2.write(l+" "+str(as_int)+"\n")
f1.close()
f2.close()