augment and integration with tidyroc

[igordot]

I am trying to pipe rpart output into the tidyroc package, but it relies on broom::augment().

The example pipe is:

glm(am ~ disp, 
  family = binomial,
  data = mtcars
  ) %>%
  augment() %>%
  make_roc(predictor = .fitted, known_class = am) %>%
  ggplot(aes(x = fpr, y = tpr)) + 
  geom_line()

Since all the tree based tidiers have been moved to broomstick, I tried that. Unfortunately, while broomstick::tidy() works, broomstick::augment() does not.

I was able to manually generate the required data frame, but is there a reason augment() is missing or is it just not ready yet?

[njtierney]

Hi @igordot - augment for rpart hasn't been implemented yet - I would be very happy to review a pull request / or other code if you would like to share what you wrote?