speed up training_model, improve accuracy