1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.app;
import java.util.ArrayList;
import java.util.List;
import weka.core.matrix.LinearRegression;
import weka.core.matrix.Matrix;
public class Test {
public static void main(String[] args) {
linearRegressionTest();
}
public static void linearRegressionTest() {
double[][] x = {{1,1,2,3}, {1,2,3,4}, {1,3,5,7}, {1,1,5,7}};
double[] y = {10, 13, 19, 17};
double[] valueToPredict = {1,1,1,1};
Matrix m = new Matrix(x);
Matrix n = new Matrix(y, y.length);
LinearRegression regression = new LinearRegression(m, n, 0);
double[] coef = regression.getCoefficients();
//predict
double a = 0;
for(int i = 0; i < coef.length; i++) {
a += coef[i] * valueToPredict[i];
}
System.out.println(a);
}
}
|