Impact of data balancing during training for best predictions