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 @@
+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);
+        }
+}