如何解决如何在 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 的结果。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。