I got 55000-size data, and I want to put them in train model on keras. But the number of train datasets are far less than original data size, which is 55000. the code and re