Improve run-time performance of regression tree training