在java中使用通用數學庫

Question

我是java的新手，現在我想將普通線性回歸應用於兩個系列，比如[1,2,3,4,5]和[2,3,4,5,6]。

我了解到有一個名為common math的庫。 但是，文檔很難理解，有沒有例子可以在java中做簡單的普通線性回歸？

Answer 1

使用math3庫，您可以執行以下操作。 示例基於SimpleRegression類：

import org.apache.commons.math3.stat.regression.SimpleRegression;

public class Try_Regression {

    public static void main(String[] args) {

        // creating regression object, passing true to have intercept term
        SimpleRegression simpleRegression = new SimpleRegression(true);

        // passing data to the model
        // model will be fitted automatically by the class 
        simpleRegression.addData(new double[][] {
                {1, 2},
                {2, 3},
                {3, 4},
                {4, 5},
                {5, 6}
        });

        // querying for model parameters
        System.out.println("slope = " + simpleRegression.getSlope());
        System.out.println("intercept = " + simpleRegression.getIntercept());

        // trying to run model for unknown data
        System.out.println("prediction for 1.5 = " + simpleRegression.predict(1.5));

    }

}