Site icon R-bloggers

Plotting Training and Testing Predictions with tidyAML

[This article was first published on Steve's Data Tips and Tricks, and kindly contributed to R-bloggers]. (You can report issue about the content on this page here)
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.
< section id="introduction" class="level1">

Introduction

In the realm of machine learning, visualizing model predictions is essential for understanding the performance and behavior of our algorithms. When it comes to regression tasks, plotting predictions alongside actual values provides valuable insights into how well our model is capturing the underlying patterns in the data. With the plot_regression_predictions() function in tidyAML, this process becomes seamless and informative.

< section id="introducing-plot_regression_predictions" class="level1">

Introducing plot_regression_predictions()

The plot_regression_predictions() function is a powerful tool for visualizing regression predictions in R. Developed as part of the tidyAML package, it leverages the capabilities of ggplot2 to create insightful plots that compare actual values with model predictions, both for training and testing datasets.

< section id="syntax-and-arguments" class="level1">

Syntax and Arguments

Let’s break down the syntax and arguments of plot_regression_predictions():

plot_regression_predictions(.data, .output = "list")
< section id="example-usage" class="level1">

Example Usage

To illustrate how plot_regression_predictions() works in practice, let’s consider an example using the mtcars dataset and a simple linear regression model.

library(tidyAML)
library(recipes)

# Define the recipe
rec_obj <- recipe(mpg ~ ., data = mtcars)

# Train the model
frt_tbl <- fast_regression(
  mtcars,
  rec_obj
)
 Setting default kernel parameters  
 Setting default kernel parameters  

In this example, we’ve created a recipe for predicting mpg based on other variables in the mtcars dataset. We then trained a fast regression model using fast_regression() from the recipes package.

Now, let’s use extract_wflw_pred() to extract the predictions:

# Extract predictions
preds <- extract_wflw_pred(frt_tbl, 1:nrow(frt_tbl))
head(preds)
# A tibble: 6 × 4
  .model_type     .data_category .data_type .value
  <chr>           <chr>          <chr>       <dbl>
1 lm - linear_reg actual         actual       15.2
2 lm - linear_reg actual         actual       30.4
3 lm - linear_reg actual         actual       21.4
4 lm - linear_reg actual         actual       33.9
5 lm - linear_reg actual         actual       19.7
6 lm - linear_reg actual         actual       10.4
unique(preds$.model_type)
 [1] "lm - linear_reg"            "brulee - linear_reg"       
 [3] "glm - linear_reg"           "stan - linear_reg"         
 [5] "dbarts - bart"              "xgboost - boost_tree"      
 [7] "rpart - decision_tree"      "earth - mars"              
 [9] "nnet - mlp"                 "brulee - mlp"              
[11] "kknn - nearest_neighbor"    "ranger - rand_forest"      
[13] "randomForest - rand_forest" "LiblineaR - svm_linear"    
[15] "kernlab - svm_linear"       "kernlab - svm_poly"        
[17] "kernlab - svm_rbf"         

With the predictions extracted, we can now plot the regression predictions using plot_regression_predictions():

# Plot regression predictions
extract_wflw_pred(frt_tbl, 1:6) |>
  plot_regression_predictions(.output = "facet")

extract_wflw_pred(frt_tbl, 1:6) |>
  plot_regression_predictions(.output = "list")
[[1]]

[[2]]

[[3]]

[[4]]

[[5]]

[[6]]

This will generate a set of plots comparing actual values with model predictions for both the training and testing datasets.

< section id="interpreting-the-plots" class="level1">

Interpreting the Plots

The plots produced by plot_regression_predictions() offer valuable insights into the performance of our regression model. Here’s what you can expect to see:

< section id="conclusion" class="level1">

Conclusion

In summary, plot_regression_predictions() is a valuable tool for visualizing regression predictions in R. Whether you’re assessing model performance, diagnosing errors, or communicating results to stakeholders, these plots provide a clear and intuitive way to understand how well your model is capturing the underlying patterns in the data. So next time you’re working on a regression task with tidyAML, don’t forget to leverage the power of visualization with plot_regression_predictions()!

To leave a comment for the author, please follow the link and comment on their blog: Steve's Data Tips and Tricks.

R-bloggers.com offers daily e-mail updates about R news and tutorials about learning R and many other topics. Click here if you're looking to post or find an R/data-science job.
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.
Exit mobile version