f_plot.R

library(data.table)
library(ROCR)
library(ggplot2)

f_plot  <- function(predictions, labels) {
    if (is.list(predictions) && is.atomic(labels))
        labels <- rep.int(list(labels), length(predictions))
    p <- prediction(predictions, labels)
    
    # Labels
    nm <- names(predictions)
    if (is.null(nm)) 
        nm <- make.names(seq_along(predictions))
    
    vals <- performance(p, "f")
    data <- data.table(
        name = rep.int(nm, lengths([email protected])),
        y = unlist([email protected]),
        x = unlist([email protected])
    )
    ggplot(data, aes(x = x, y = y, color = name)) + 
        geom_line() +
        labs(y = [email protected],
             x = [email protected],
             color = "") +
        facet_wrap(~ name, ncol = 1L, scales = "free") +
        theme(legend.position = "bottom")
}

pr_plot.R

library(data.table)
library(ROCR)
library(ggplot2)

pr_plot <- function(predictions, labels) {
    if (is.list(predictions) && is.atomic(labels))
        labels <- rep.int(list(labels), length(predictions))
    p <- prediction(predictions, labels)
    
    # Labels
    nm <- names(predictions)
    if (is.null(nm)) 
        nm <- make.names(seq_along(predictions))
    
    vals <- performance(p, "prec", "rec")
    data <- data.table(
        name = rep.int(nm, lengths([email protected])),
        y1 = unlist([email protected]),
        y2 = unlist([email protected]),
        x = unlist([email protected])
    )
    setnames(data, c("y1", "y2"), c([email protected], [email protected]))
    data <- melt(data, id.vars = c("x", "name"))
    ggplot(data, aes(x = x, y = value, color = variable)) + 
        geom_line() +
        labs(y = "Value",
             x = [email protected],
             color = "") +
        facet_wrap(~ name, ncol = 1L, scales = "free") +
        theme(legend.position = "bottom")
}

roc_plot.R

library(data.table)
library(ROCR)
library(ggplot2)

roc_plot <- function(predictions, labels) {
    if (is.list(predictions) && is.atomic(labels))
        labels <- rep.int(list(labels), length(predictions))
    
    p <- prediction(predictions, labels)
    
    # AUC
    auc <- performance(p, "auc")
    auc <- unlist([email protected])
    
    # Labels
    nm <- names(predictions)
    if (is.null(nm)) 
        nm <- make.names(seq_along(predictions))
    nm <- sprintf("%s: AUC = %.3f", nm, auc)
    
    # ROC
    vals <- performance(p, "tpr", "fpr")
    data <- data.table(
        name = rep.int(nm, lengths([email protected])),
        x = unlist([email protected]),
        y = unlist([email protected])
    )
    
    # Plot
    ggplot(data, aes(x = x, y = y, color = name)) + 
        geom_path() + 
        labs(x = [email protected],
             y = [email protected],
             color = "") +
        theme(legend.position = "bottom",
              legend.direction = "vertical")
}