线性回归算法Java代码
时间: 2023-12-23 07:44:49 浏览: 46
这是一个基本的线性回归算法Java代码示例:
```java
import java.util.Arrays;
public class LinearRegression {
private double[] x;
private double[] y;
private double slope;
private double intercept;
public LinearRegression(double[] x, double[] y) {
if (x.length != y.length) {
throw new IllegalArgumentException("Arrays must be of equal length.");
}
this.x = x;
this.y = y;
calculate();
}
public double getSlope() {
return slope;
}
public double getIntercept() {
return intercept;
}
private void calculate() {
int n = x.length;
double sumX = 0.0;
double sumY = 0.0;
double sumXY = 0.0;
double sumXX = 0.0;
for (int i = 0; i < n; i++) {
sumX += x[i];
sumY += y[i];
sumXY += x[i] * y[i];
sumXX += x[i] * x[i];
}
double xMean = sumX / n;
double yMean = sumY / n;
slope = (sumXY - n * xMean * yMean) / (sumXX - n * xMean * xMean);
intercept = yMean - slope * xMean;
}
public static void main(String[] args) {
double[] x = {1, 2, 3, 4, 5};
double[] y = {2.1, 3.9, 6.1, 8.2, 10.1};
LinearRegression lr = new LinearRegression(x, y);
System.out.println("Slope: " + lr.getSlope());
System.out.println("Intercept: " + lr.getIntercept());
}
}
```
这段代码实现了一个基本的线性回归算法,它接受两个数组作为输入,一个包含自变量的值,另一个包含因变量的值。它使用这些值来计算回归系数和截距。在这个例子中,我们使用了一个简单的数据集,并输出了回归系数和截距的值。
相关推荐
![doc](https://img-home.csdnimg.cn/images/20210720083327.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)