R/residuals_plot.R

In John-R-Wallace-NOAA/FishNIRS: NIRS on fish structures.

residuals_plot <- function(observed_val, predicted_val, model_name = "", ...) {

  require(ggplot2)
  
  residual_val <- observed_val - predicted_val
  RMSE <- signif(sqrt(mean((observed_val - predicted_val)^2, na.rm = TRUE)), 6)
  MAE <- signif(mean(abs(observed_val - predicted_val), na.rm = TRUE), 6)

  ggplot(mapping = aes(x = predicted_val, y = residual_val, col = abs(residual_val))) +
  geom_point(alpha = 0.9, size = 2) +
  geom_abline(intercept = 0, slope = 0) +
    # facet_wrap(~) +
    labs(
       title = paste0(model_name, "\nResiduals: Test Set"),
       subtitle = paste0("RMSE: ", RMSE, ", MAE: ", round(MAE, 3)),
       x = "Predicted",
       y = "Residual (Obs - Pred)",
       col = "Absolute Deviation"
       ) +
  theme_bw() +
  theme(aspect.ratio = 0.9, panel.grid.minor.x = element_blank(), legend.title = element_text(size = 10, face="bold"), 
          legend.text = element_text(size = 9), plot.title = element_text(size=12, face="bold"), axis.title=element_text(size=10, face="bold"), 
          axis.text.x = element_text(angle = 0), legend.position = "none") +
  # scale_x_continuous(expand = c(0,0)) +
  # scale_y_continuous(expand = c(0,0)) +
  coord_equal() + scale_color_viridis_c(direction = -1)
}