R/g_autoplot.R

In precrec: Calculate Accurate Precision-Recall and ROC (Receiver Operator Characteristics) Curves

#
#  Plot score distributions by rank
#
autoplot.fmdat <- function(object, ...) {
  curve_df <- .prepare_autoplot(object)

  # === Create a ggplot object ===
  x_col <- rlang::sym("x")
  y_col <- rlang::sym("y")
  p <- ggplot2::ggplot(
    curve_df,
    ggplot2::aes(x = !!x_col, y = !!y_col, color = !!x_col)
  )
  p <- p + ggplot2::geom_jitter()
  p <- p + ggplot2::coord_flip()
  p <- .geom_basic(p, "Score distributions by rank", NULL, "rank",
    show_legend = FALSE
  )
}

#
# Plot TPs, FNs, FPs, TNs by ranks
#
autoplot.cmats <- function(object, ...) {
  curve_df <- .prepare_autoplot(object)

  # === Create a ggplot object ===
  x_col <- rlang::sym("x")
  y_col <- rlang::sym("y")
  group_col <- rlang::sym("group")
  p <- ggplot2::ggplot(
    curve_df,
    ggplot2::aes(x = !!x_col, y = !!y_col, color = !!group_col)
  )
  p <- p + ggplot2::geom_line()
  p <- .geom_basic(p, "TPs, FNs, FPs, and TNs by ranks",
    "rank", "count",
    show_legend = TRUE
  )
}

#
# Plot basic evaluation measures by rank
#
autoplot.pevals <- function(object, ...) {
  curve_df <- .prepare_autoplot(object)

  # === Create a ggplot object ===
  x_col <- rlang::sym("x")
  y_col <- rlang::sym("y")
  p <- ggplot2::ggplot(
    curve_df,
    ggplot2::aes(x = !!x_col, y = !!y_col)
  )
  p <- p + ggplot2::geom_line()
  p <- p + ggplot2::facet_wrap(~group, ncol = 2)

  p <- .geom_basic(p, "Evaluation measures by ranks",
    "normalized rank", "evaluation value",
    show_legend = FALSE
  )
}

#' @rdname autoplot
#' @export
autoplot.sscurves <- function(object, curvetype = c("ROC", "PRC"), ...) {
  arglist <- .get_autoplot_arglist(attr(object, "args"),
    def_curvetype = curvetype, def_type = "l",
    def_show_cb = FALSE, def_raw_curves = TRUE,
    def_add_np_nn = TRUE,
    def_show_legend = FALSE,
    def_ret_grob = FALSE,
    def_reduce_points = TRUE,
    def_multiplot_lib = "patchwork", ...
  )
  arglist[["raw_curves"]] <- TRUE
  arglist[["show_cb"]] <- FALSE
  arglist[["curvetype"]] <- .pmatch_curvetype_rocprc(arglist[["curvetype"]])

  .autoplot_multi(object, arglist)
}

#' @rdname autoplot
#' @export
autoplot.mscurves <- function(object, curvetype = c("ROC", "PRC"), ...) {
  arglist <- .get_autoplot_arglist(attr(object, "args"),
    def_curvetype = curvetype, def_type = "l",
    def_show_cb = FALSE, def_raw_curves = TRUE,
    def_add_np_nn = TRUE,
    def_show_legend = TRUE,
    def_ret_grob = FALSE,
    def_reduce_points = TRUE,
    def_multiplot_lib = "patchwork", ...
  )
  arglist[["raw_curves"]] <- TRUE
  arglist[["show_cb"]] <- FALSE
  arglist[["curvetype"]] <- .pmatch_curvetype_rocprc(arglist[["curvetype"]])

  .autoplot_multi(object, arglist)
}

#' @rdname autoplot
#' @export
autoplot.smcurves <- function(object, curvetype = c("ROC", "PRC"), ...) {
  arglist <- .get_autoplot_arglist(attr(object, "args"),
    def_curvetype = curvetype, def_type = "l",
    def_show_cb = TRUE, def_raw_curves = NULL,
    def_add_np_nn = TRUE,
    def_show_legend = FALSE,
    def_ret_grob = FALSE,
    def_reduce_points = TRUE,
    def_multiplot_lib = "patchwork", ...
  )

  arglist[["curvetype"]] <- .pmatch_curvetype_rocprc(arglist[["curvetype"]])

  .autoplot_multi(object, arglist)
}

#' @rdname autoplot
#' @export
autoplot.mmcurves <- function(object, curvetype = c("ROC", "PRC"), ...) {
  arglist <- .get_autoplot_arglist(attr(object, "args"),
    def_curvetype = curvetype, def_type = "l",
    def_show_cb = FALSE, def_raw_curves = NULL,
    def_add_np_nn = TRUE,
    def_show_legend = TRUE,
    def_ret_grob = FALSE,
    def_reduce_points = TRUE,
    def_multiplot_lib = "patchwork", ...
  )

  arglist[["curvetype"]] <- .pmatch_curvetype_rocprc(arglist[["curvetype"]])

  .autoplot_multi(object, arglist)
}

#' @rdname autoplot
#' @export
autoplot.sspoints <- function(object, curvetype = .get_metric_names("basic"),
                              ...) {
  arglist <- .get_autoplot_arglist(attr(object, "args"),
    def_curvetype = curvetype, def_type = "p",
    def_show_cb = FALSE, def_raw_curves = TRUE,
    def_add_np_nn = TRUE,
    def_show_legend = FALSE,
    def_ret_grob = FALSE,
    def_reduce_points = FALSE,
    def_multiplot_lib = "patchwork", ...
  )
  arglist[["raw_curves"]] <- TRUE
  arglist[["show_cb"]] <- FALSE
  arglist[["curvetype"]] <- .pmatch_curvetype_basic(arglist[["curvetype"]])

  .autoplot_multi(object, arglist)
}

#' @rdname autoplot
#' @export
autoplot.mspoints <- function(object, curvetype = .get_metric_names("basic"),
                              ...) {
  arglist <- .get_autoplot_arglist(attr(object, "args"),
    def_curvetype = curvetype, def_type = "p",
    def_show_cb = FALSE, def_raw_curves = TRUE,
    def_add_np_nn = TRUE,
    def_show_legend = TRUE,
    def_ret_grob = FALSE,
    def_reduce_points = FALSE,
    def_multiplot_lib = "patchwork", ...
  )
  arglist[["raw_curves"]] <- TRUE
  arglist[["show_cb"]] <- FALSE
  arglist[["curvetype"]] <- .pmatch_curvetype_basic(arglist[["curvetype"]])

  .autoplot_multi(object, arglist)
}

#' @rdname autoplot
#' @export
autoplot.smpoints <- function(object, curvetype = .get_metric_names("basic"),
                              ...) {
  arglist <- .get_autoplot_arglist(attr(object, "args"),
    def_curvetype = curvetype, def_type = "p",
    def_show_cb = TRUE, def_raw_curves = NULL,
    def_add_np_nn = TRUE,
    def_show_legend = FALSE,
    def_ret_grob = FALSE,
    def_reduce_points = FALSE,
    def_multiplot_lib = "patchwork", ...
  )

  arglist[["curvetype"]] <- .pmatch_curvetype_basic(arglist[["curvetype"]])

  .autoplot_multi(object, arglist)
}

#' @rdname autoplot
#' @export
autoplot.mmpoints <- function(object, curvetype = .get_metric_names("basic"),
                              ...) {
  arglist <- .get_autoplot_arglist(attr(object, "args"),
    def_curvetype = curvetype, def_type = "p",
    def_show_cb = FALSE, def_raw_curves = NULL,
    def_add_np_nn = TRUE,
    def_show_legend = TRUE,
    def_ret_grob = FALSE,
    def_reduce_points = FALSE,
    def_multiplot_lib = "patchwork", ...
  )

  arglist[["curvetype"]] <- .pmatch_curvetype_basic(arglist[["curvetype"]])

  .autoplot_multi(object, arglist)
}