model_formula <- reformulate(features, response = "response")

model <- training_data %>%
  ml_gradient_boosted_trees(formula)

predicted <- ml_predict(
    model,
    testing_data) %>% pull(prediction)

results <- testing_data %>%
  select(response)
  collect() %>%
  mutate(predicted_response = predicted)