in ttw/train/classify_landmarks.py [0:0]
def create_split(dataset):
keys = dataset[0].keys()
train_data = {k: list() for k in keys}
valid_data = {k: list() for k in keys}
for i in range(len(dataset)):
if random.random() > 0.7:
for k in dataset[i].keys():
valid_data[k].append(dataset[i][k])
else:
for k in dataset[i].keys():
train_data[k].append(dataset[i][k])
return train_data, valid_data