Takes in a vector of prediction labels and a vector of prediction probabilities.

plot_roc(y_label, predict_proba)

Arguments

y_label

a numeric vector contains the validation set label, all elements should be either 0 or 1

predict_proba

a numberic vector contains the prediction probabilities

Value

A plot

Examples

plot_roc(c(1,0,0,1,1), c(0.5,0.34,0.89,0.12,0.99))
#> Setting levels: control = 0, case = 1
#> Setting direction: controls > cases