@discourse-admin
There are only 4 images in data with classes [1, 2] combined. Is there any chances for any other permutations of classes in test and final sets.
Hi @ArbaazBeg
The hold out dataset has the same combinations of classes as the training dataset.
good luck!
ThinkOnward Team