Training the Keras model for a dataset of size 6000 but it is only getting trained on 188 [duplicate]