Question

这是一个与我之前的问题geom_smooth with facet_grid and different fitting functions有关的问题。在那个问题中，我试图在geom_smooth中为ggplot2的facet网格中的每个facet使用不同的拟合函数。 Marco Sandri友好地提供了一个答案，我正在尝试使用用户定义的公式而不是现有的公式（例如lm，loess）。这是我的代码。

# Load library
library(ggplot2)

# Load data
data(mtcars)

# Smoothing function with different behaviour depending on the panel
custom.smooth <- function(formula, data,...){
  smooth.call <- match.call()

  if(as.numeric(unique(data$PANEL)) == 6) {
    # Nonlinear regression
    method.name <- eval(parse(text="nls"))
    # Specify formula
    formula <- as.formula("y ~ a * x^b")
    # Add initial parameters
    smooth.call[["start"]] <- c(a = 10, b = -0.5)
  }else{
    # Linear regression
    method.name <- eval(parse(text="lm"))
  }

  # Add function name
  smooth.call[[1]] <- method.name
  # Perform fit
  eval.parent(smooth.call)
}

# Plot data with custom fitting function
p <- ggplot(mtcars,aes(x = disp, y = mpg)) + geom_point() + facet_grid(gear ~ am)
p <- p + geom_smooth(method = "custom.smooth")
print(p)

在这段代码中，我定义了一个函数custom.smooth，它选择了适合的模型。在此示例中，除了面板6之外，所有模型都是线性回归，而面板6是用户定义的函数y ~ a*x^b。运行此代码会出现错误：

警告消息：stat_smooth()中的计算失败：奇异初始参数估计时的梯度矩阵

然而，当我使用这些初始参数对面板6中的数据运行nls时，我没有得到这样的错误（即nls(mpg ~ a * disp^b, mtcars %>% filter(gear == 5, am == 1), start = c(a = 10, b = -0.5))）。这让我觉得nls没有看到我指定的起始值。我也尝试在geom_smooth函数中指定这些参数，如下所示：

p <- p + geom_smooth(method = "custom.smooth", method.args = list(start = c(a = 10, b = -0.5)))

但我遇到了同样的问题。我有什么想法可以将我的起始值设为nls？或者是否有其他原因导致代码无效？

Answer 1

这是解决方案，它从this post中受益匪浅。我不知道为什么以前的版本不起作用，但这似乎工作正常。

# Load library
library(ggplot2)

# Load data
data(mtcars)

# Smoothing function with different behaviour depending on the panel
custom.smooth <- function(formula, data,...){
  smooth.call <- match.call()

  if(as.numeric(unique(data$PANEL)) == 6) {
    # Nonlinear regression
    smooth.call[[1]] <- quote(nls)
    # Specify formula
    smooth.call$formula <- as.formula("y ~ a * x ^ b")
    # Add initial parameters
    smooth.call$start <- c(a = 300, b = -0.5)
  }else{
    # Linear regression
    smooth.call[[1]] <- quote(lm)
  }

  # Perform fit
  eval.parent(smooth.call)
}

# Plot data with custom fitting function
p <- ggplot(mtcars,aes(x = disp, y = mpg)) + geom_point() + facet_grid(gear ~ am)
p <- p + geom_smooth(method = "custom.smooth", se = FALSE)
print(p)

R：在geom_smooth中的自定义函数中使用时，不会获取其他参数

1 个答案: