java求多项式回归,使用Apache Maths（Java）进行多项式回归

Could anybody help me make a polynomial regression (order 2) with the Apache Math library.

The following data should give this equation: 39.79 x^2 - 497.66 x + 997.45

(computed by Excel with 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));

Here are the regression coefficients provided by the previous code:

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

Obviously, there's something I'm missing...

Thanks for your kind help

Dom

解决方案

All your data points are at x = 1.

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

instead of 1.0 there should be x value, if they are evenly spaced from zero than use for loop and ix instead of 1.0.