带输出图和回归线的梯度下降函数

Question

我一直在运行以下代码，它返回正确的系数。 但是，无论我在哪里放置plot调用，都无法获得任何绘图输出。

我不确定这里是否需要一个可重现的例子，因为我认为这可以通过查看下面的gradientDescent下降函数来解决？ 这是我第一次尝试在 R 中运行这个算法：

gradientDescent <- function(x, y, learn_rate, conv_threshold, n, max_iter) {
  m <- runif(1, 0, 1)
  c <- runif(1, 0, 1)
  yhat <- m * x + c
  cost_error <- (1 / (n + 2)) * sum((y - yhat) ^ 2)
  converged = F
  iterations = 0
  while(converged == F) {
    m_new <- m - learn_rate * ((1 / n) * (sum((yhat - y) * x)))
    c_new <- c - learn_rate * ((1 / n) * (sum(yhat - y)))
    m <- m_new
    c <- c_new
    yhat <- m * x + c
    cost_error_new <- (1 / (n + 2)) * sum((y - yhat) ^ 2)
    if(cost_error - cost_error_new <= conv_threshold) {
      converged = T
    }
    iterations = iterations + 1
    if(iterations > max_iter) {
      converged = T
    return(paste("Optimal intercept:", c, "Optimal slope:", m))
    }
  }
}

Answer 1

目前还不清楚你一直在做什么是无效的。 即使在函数内部使用，基本图形函数plot和abline应该能够产生输出。 Lattice 和 ggplot2 图形基于grid -grpahics，因此需要在函数调用周围使用print()来创建输出（如 R-FAQ 中所述）。 所以试试这个：

gradientDescent <- function(x, y, learn_rate, conv_threshold, n, max_iter) 
    { ## plot.new() perhaps not needed
      plot(x,y)
      m <- runif(1, 0, 1)
      c <- runif(1, 0, 1)
      yhat <- m * x + c
      cost_error <- (1 / (n + 2)) * sum((y - yhat) ^ 2)
      converged = F
      iterations = 0
      while(converged == F) {
        m_new <- m - learn_rate * ((1 / n) * (sum((yhat - y) * x)))
        c_new <- c - learn_rate * ((1 / n) * (sum(yhat - y)))
        m <- m_new
        c <- c_new
        yhat <- m * x + c
        cost_error_new <- (1 / (n + 2)) * sum((y - yhat) ^ 2)
        if(cost_error - cost_error_new <= conv_threshold) {
          converged = T
        }
        iterations = iterations + 1
        if(iterations > max_iter) { abline( c, m)  #calculated 
          dev.off()
          converged = T
        return(paste("Optimal intercept:", c, "Optimal slope:", m))
        }
      }
    }