java - 使用 Apache Maths (Java) 进行多项式回归

Question

谁能帮我用 Apache 数学库做一个多项式回归(2 阶)。

以下数据应给出此等式:39.79 x^2 - 497.66 x + 997.45(用Excel计算，r2 = 0.9998)

// coding style from http://commons.apache.org/proper/commons-math/userguide/fitting.html    

double[] y = { 540.0, 160.0, -140.0, -360.0, -480.0, -560.0, -540.0, -440.0, -260.0, 0.0, 340.0};              
        final WeightedObservedPoints obs = new WeightedObservedPoints();
        for (double figure:y){
            obs.add(1.0, figure);
        }
        final PolynomialCurveFitter fitter = PolynomialCurveFitter.create(2);
        final double[] coeff = fitter.fit(obs.toList());
        System.out.println("coef="+Arrays.toString(coeff));

这里是前面代码提供的回归系数:

coef=[-53.73522460839947, -52.22329678670934, -52.22329678670934]

显然，我缺少一些东西......

感谢您的帮助

主场

Answer 1

您所有的数据点都在 x = 1 处。

obs.add(1.0, figure);!!!!

应该有 x 值而不是 1.0，如果它们与零间隔均匀而不是使用 for 循环和 ix 而不是 1.0。