Select specific MNIST classes to train a neural network in TensorFlow
Found the answer i guess... one hot=True transformed the scalar into a one hot vector :) thanks for your time anyway!
currently i am looking for a way to filter specific classes out of my training dataset (MNIST) to train a neural network on different constellations, e.g. train a network only on classes 4,5,6 then train it on 0,1,2,3,4,5,6,7,8,9 to evaluate the results with the test dataset.
I'd like to do it with an argument parser via console to chose which classes should be in my training dataset so i can split this into mini batches. I think i could do it with sorting out via labels but i am kinda stuck at this moment... would appreciate any tip!!!
Greetings,
Alex