plot_train_vs_test: plot_train_vs_test

View source: R/ranger_reg_plot.R

plot_train_vs_testR Documentation

plot_train_vs_test

Description

Plot the residuals of observed and predicted values from a ranger model.

Usage

plot_train_vs_test(
  train_y,
  predicted_train_y,
  test_y,
  predicted_test_y,
  train_SampleIDs = NULL,
  test_SampleIDs = NULL,
  train_prefix = "train",
  test_prefix = "test",
  train_target_field,
  test_target_field,
  outdir = NULL
)

Arguments

train_y

The numeric labels for training data.

predicted_train_y

The predicted values for training data.

test_y

The numeric labels for testing data.

predicted_test_y

The predicted values for test data.

train_SampleIDs

The sample ids in the train data.

test_SampleIDs

The sample ids in the test data.

train_prefix

The prefix for the dataset in the training data.

test_prefix

The prefix for the dataset in the testing data.

train_target_field

A string indicating the target field of the training metadata for regression.

test_target_field

A string indicating the target field of the testing metadata for regression.

outdir

The output directory.

Author(s)

Shi Huang

Examples

set.seed(123)
x <- data.frame(rbind(t(rmultinom(7, 75, c(.201,.5,.02,.18,.099))),
            t(rmultinom(8, 75, c(.201,.4,.12,.18,.099))),
            t(rmultinom(15, 75, c(.011,.3,.22,.18,.289))),
            t(rmultinom(15, 75, c(.091,.2,.32,.18,.209))),
            t(rmultinom(15, 75, c(.001,.1,.42,.18,.299)))))
y<- 1:60
rf_model<-rf.out.of.bag(x, y)
plot_perf_VS_rand(y=y, predicted_y=rf_model$predicted, prefix="train",
permutation=100, metric="MAE", target_field="age")

shihuang047/crossRanger documentation built on Feb. 7, 2023, 10:03 p.m.