use train method to prepare comparison for prediction