diff options
author | 2019-06-24 10:12:34 -0400 | |
---|---|---|
committer | 2019-06-24 10:12:34 -0400 | |
commit | 69729ebb20fc34f5d836d0ba9dc416114f2c9c4a (patch) | |
tree | 7e6a9d3ce47756514f232ff23476a5d0be588990 /Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/app/Test.java | |
parent | estimate value function with simple linear regression (diff) | |
download | VIATRA-Generator-69729ebb20fc34f5d836d0ba9dc416114f2c9c4a.tar.gz VIATRA-Generator-69729ebb20fc34f5d836d0ba9dc416114f2c9c4a.tar.zst VIATRA-Generator-69729ebb20fc34f5d836d0ba9dc416114f2c9c4a.zip |
Implement linear regressor using Weka3
Diffstat (limited to 'Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/app/Test.java')
-rw-r--r-- | Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/app/Test.java | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/app/Test.java b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/app/Test.java new file mode 100644 index 00000000..f06b377f --- /dev/null +++ b/Metrics/Metrics-Calculation/ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator/src/ca/mcgill/ecse/dslreasoner/realistic/metrics/calculator/app/Test.java | |||
@@ -0,0 +1,31 @@ | |||
1 | package ca.mcgill.ecse.dslreasoner.realistic.metrics.calculator.app; | ||
2 | |||
3 | import java.util.ArrayList; | ||
4 | import java.util.List; | ||
5 | |||
6 | import weka.core.matrix.LinearRegression; | ||
7 | import weka.core.matrix.Matrix; | ||
8 | |||
9 | public class Test { | ||
10 | public static void main(String[] args) { | ||
11 | linearRegressionTest(); | ||
12 | } | ||
13 | |||
14 | public static void linearRegressionTest() { | ||
15 | double[][] x = {{1,1,2,3}, {1,2,3,4}, {1,3,5,7}, {1,1,5,7}}; | ||
16 | double[] y = {10, 13, 19, 17}; | ||
17 | double[] valueToPredict = {1,1,1,1}; | ||
18 | Matrix m = new Matrix(x); | ||
19 | Matrix n = new Matrix(y, y.length); | ||
20 | |||
21 | LinearRegression regression = new LinearRegression(m, n, 0); | ||
22 | double[] coef = regression.getCoefficients(); | ||
23 | |||
24 | //predict | ||
25 | double a = 0; | ||
26 | for(int i = 0; i < coef.length; i++) { | ||
27 | a += coef[i] * valueToPredict[i]; | ||
28 | } | ||
29 | System.out.println(a); | ||
30 | } | ||
31 | } | ||