Draws train vs test RMSE plot for models

Draws train vs test RMSE plot for models

draw_rmse_plot(
  train_observed,
  train_predictions,
  test_observed,
  test_predictions,
  models
)

Arguments

train_observed

A vector of the true values for training dataset.

train_predictions

A list of lists of predictions on the training dataset for models but just for c('ranger', 'xgboost', 'decision_tree')

test_observed

A vector of the true values for test dataset.

test_predictions

A list of lists of predictions on test dataset for models but just for c('ranger', 'xgboost', 'decision_tree')

models

A list of models.