Skip to contents

Create ggplot2 plot of regression predictions

Source: R/plot-regression-predictions.R
plot_regression_predictions.Rd

Create a ggplot2 plot of regression predictions.

Usage

plot_regression_predictions(.data, .output = "list")

Arguments

.data

The data from the output of the extract_regression_residuals() function.

.output

The default is "list" which will return a list of plots. The other option is "facet" which will return a single faceted plot.

Value

A list of ggplot2 plots or a faceted plot.

Details

Create a ggplot2 plot of regression predictions, the actual, training, and testing values. The output of this function can either be a list of plots or a single faceted plot. This function takes the output of the function extract_wflw_pred() function.

See also

Other Plotting: plot_regression_residuals()

Author

Steven P. Sanderson II, MPH

Examples

library(recipes)

rec_obj <- recipe(mpg ~ ., data = mtcars)
frt_tbl <- fast_regression(
  mtcars,
  rec_obj,
  .parsnip_eng = c("lm","glm"),
  .parsnip_fns = "linear_reg"
  )

extract_wflw_pred(frt_tbl,1) |> plot_regression_predictions()
#> [[1]]
#> Warning: Removed 32 rows containing missing values or values outside the scale range
#> (`geom_line()`).
#> Warning: Removed 40 rows containing missing values or values outside the scale range
#> (`geom_line()`).
#> Warning: Removed 56 rows containing missing values or values outside the scale range
#> (`geom_line()`).

#> 
extract_wflw_pred(frt_tbl,1:nrow(frt_tbl)) |>
  plot_regression_predictions(.output = "facet")