The CNN model will classify between two class with training samples = 5974 and validation = 1987.
I am using datagen.flow_from_directory