7

我正在尝试将stat_ecdf()累积成功绘制为预测模型创建的排名分数的函数。

#libraries
require(ggplot2)
require(scales)

# fake data for reproducibility
set.seed(123)
n <- 200
df <- data.frame(model_score= rexp(n=n,rate=1:n),
                 obs_set= sample(c("training","validation"),n,replace=TRUE))
df$model_rank <- rank(df$model_score)/n
df$target_outcome <- rbinom(n,1,1-df$model_rank)

# Plot Gain Chart using stat_ecdf()
ggplot(subset(df,target_outcome==1),aes(x = model_rank)) + 
  stat_ecdf(aes(colour = obs_set), size=1) + 
  scale_x_continuous(limits=c(0,1), labels=percent,breaks=seq(0,1,.1)) +
  xlab("Model Percentile") + ylab("Percent of Target Outcome") +
  scale_y_continuous(limits=c(0,1), labels=percent) +
  geom_segment(aes(x=0,y=0,xend=1,yend=1), 
               colour = "gray", linetype="longdash", size=1) +
  ggtitle("Gain Chart")

在此处输入图像描述

我要做的就是强制 ECDF 从 (0,0) 开始并在 (1,1) 结束，这样曲线的开头或结尾就没有间隙。如果可能的话，我想在 的语法中执行它ggplot2，但我会满足于一个聪明的解决方法。

@Henrik 这不是这个问题的重复，因为我已经用 and 定义了我的限制scale_x_，_y_continuous()并且添加expand_limits()不会做任何事情。需要修复的不是 PLOT 的起点，而是 stat_ecdf() 的端点。

4

1 回答 1

1

不幸的是，stat_ecdf这里的定义没有任何回旋的余地。它在内部确定端点。

有一个有点先进的解决方案。使用最新版本的 ggplot2 ( devtools::install_github("hadley/ggplot2"))，可扩展性得到了改进，可以覆盖此行为，但并非没有一些样板文件。

stat_ecdf2 <- function(mapping = NULL, data = NULL, geom = "step",
                      position = "identity", n = NULL, show.legend = NA,
                      inherit.aes = TRUE, minval=NULL, maxval=NULL,...) {
  layer(
    data = data,
    mapping = mapping,
    stat = StatEcdf2,
    geom = geom,
    position = position,
    show.legend = show.legend,
    inherit.aes = inherit.aes,
    stat_params = list(n = n, minval=minval,maxval=maxval),
    params = list(...)
  )
}


StatEcdf2 <- ggproto("StatEcdf2", StatEcdf,
  calculate = function(data, scales, n = NULL, minval=NULL, maxval=NULL, ...) {
    df <- StatEcdf$calculate(data, scales, n, ...)
    if (!is.null(minval)) { df$x[1] <- minval }
    if (!is.null(maxval)) { df$x[length(df$x)] <- maxval }
    df
  }
)

现在，stat_ecdf2将表现与 相同stat_ecdf，但带有一个可选的minval和maxval参数。所以这可以解决问题：

ggplot(subset(df,target_outcome==1),aes(x = model_rank)) +
  stat_ecdf2(aes(colour = obs_set), size=1, minval=0, maxval=1) +
  scale_x_continuous(limits=c(0,1), labels=percent,breaks=seq(0,1,.1)) +
  xlab("Model Percentile") + ylab("Percent of Target Outcome") +
  scale_y_continuous(limits=c(0,1), labels=percent) +
  geom_segment(aes(x=0,y=0,xend=1,yend=1),
               colour = "gray", linetype="longdash", size=1) +
  ggtitle("Gain Chart")

这里最大的警告是，我不知道将来是否会支持当前的可扩展性模型；过去它已经改变了好几次，使用“ggproto”的变化是最近的——比如最近的 2015 年 7 月 15 日。

另外，这让我有机会真正深入研究 ggplot 的内部结构，这是我一段时间以来一直想做的事情。

于 2015-08-05T14:54:21.787 回答