plot_train_test: Create Test/Train AUC Plot

In jnshsrs/PEDISdata: Preprocessing PEDIS data files

Description

Create Test/Train AUC Plot

Usage

plot_train_test(trainROC, testROC)

Arguments

trainROC	A roc obj (on in-sample data)
testROC	A roc obj (on out-sample data)

Value

A plot

Examples

data(pedis)
X_y <- prepare_X_y(data = pedis, outcome = "minor_amputation", vars = c("p", "d"))
folds <- create_folds(y = X_y$y)
fit <- train_pedis(X = X_y$X, y = X_y$y, folds = folds)
out_sample_prediction <- train_predict_all(folds = folds, X_y = X_y)
in_sample_prediction <- test_predict(fit)
out_sample_auc <- create_roc(out_sample_prediction)
in_sample_auc <- create_roc(in_sample_prediction)
plot_train_test(trainROC = in_sample_auc, testROC = out_sample_auc)

jnshsrs/PEDISdata documentation built on June 24, 2019, 12:07 p.m.