javascript中两个自变量的线性回归

function linearRegression(x1, x2, y) {
    const n = x1.length;
    let sumX1 = 0, sumX2 = 0, sumY = 0;
    let sumX1Y = 0, sumX1X2 = 0, sumX2Y = 0, sumX1Sq = 0, sumX2Sq = 0;

    for (let i = 0; i < n; i++) {
        sumX1 += x1[i];
        sumX2 += x2[i];
        sumY += y[i];
        sumX1Y += x1[i] * y[i];
        sumX1X2 += x1[i] * x2[i];
        sumX2Y += x2[i] * y[i];
        sumX1Sq += x1[i] ** 2;
        sumX2Sq += x2[i] ** 2;
    }

    const det = n * (sumX1Sq * sumX2Sq - sumX1X2 ** 2) - (sumX1 * sumX2) * (sumX1 * sumX2);
    const b1 = (n * (sumX1Y * sumX2Sq - sumX1X2 * sumX2Y) - sumX1 * sumX2 * sumY + sumX1 * sumX2 * sumY) / det;
    const b2 = (n * (sumX1Sq * sumX2Y - sumX1 * sumX1Y) - sumX1 * sumX2 * sumY + sumX1 * sumX2 * sumY) / det;
    const b0 = (sumY - b1 * sumX1 - b2 * sumX2) / n;

    return { b0, b1, b2 };
}

// 示例使用
const x1 = [1, 2, 3, 4, 5];
const x2 = [5, 4, 3, 2, 1];
const y = [10, 20, 25, 30, 35];

const { b0, b1, b2 } = linearRegression(x1, x2, y);
console.log(`回归方程: y = ${b0} + ${b1}x1 + ${b2}x2`);