how can i train the Siamese network under k-shot

i needs to test a siamese network for k- shot learning

how can i determine that the network trained on k-samples from each folder to test it's performance

for example if k=5 , then the network only trained with five shots from each class

how can i do that without manually splitting the data as i needs to test several values of k

please help