如何在 JavaScript 中计算 R 平方值？

问题描述

我需要为我的回归模型计算 R 平方值（最小二乘法）。如何在 JavaScript 中执行此操作？

例如我有这个数据：

x = [5,7,8,2,17,9,4,11,12,6]
y = [99,86,87,88,111,103,94,78,77,85,86]

在做我的回归数学之后，我得到了这些系数（格式：a + bx）[103.10596026,-1.75128771] 代表线 y = 103.10596026 - 1.75128771x。

现在我想计算 R 平方值但不知道如何...示例函数模板如下所示：

function rSquared(xData,yData,modelCoefficients) {

  // Do stuff here...

  return rSquaredValue

}

谢谢！

解决方法

好的，我认为 this function 应该可以解决问题：

function rSquared(x,y,coefficients) {

  let regressionSquaredError = 0
  let totalSquaredError = 0

  function yPrediction(x,coefficients) {
    return coefficients[0] + coefficients[1] * x
  }

  let yMean = y.reduce((a,b) => a + b) / y.length

  for (let i = 0; i < x.length; i++) {
    regressionSquaredError += Math.pow(y[i] - yPrediction(x[i],coefficients),2)
    totalSquaredError += Math.pow(y[i] - yMean,2)
  }

  return 1 - (regressionSquaredError / totalSquaredError)

}

我已经在示例数据上对其进行了测试并得到了这个结果，0.5754611008553385 女巫也匹配了来自 this online calculator 的结果。