minor edits to improve efficiency
don't keep data with rf and gbm to save memory