Extract LinearRegressionTest training set

Zomis · Jan 11, 2016 · 3ca8fac · 3ca8fac
1 parent 114027c
commit 3ca8fac
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/test/groovy/net/zomis/machlearn/regression/LinearRegressionTest.groovy b/src/test/groovy/net/zomis/machlearn/regression/LinearRegressionTest.groovy
@@ -9,27 +9,23 @@ class LinearRegressionTest {
         GradientDescent.gradientDescent(LinearRegression.costFunction(xs, y),
                 new ConvergenceIterations(10000), new double[thetaParameters], 0.01)
     }
+    double[][] xs = [[1] as double[], [2] as double[], [3] as double[], [4] as double[]] as double[][]
+    double[] y = [2, 4, 5, 8] as double[]
 
     @Test(expected = IllegalArgumentException)
     void requiresParameters() {
-        double[][] xs = [[1] as double[], [2] as double[], [3] as double[], [4] as double[]] as double[][]
-        double[] y = [2, 4, 5, 8] as double[]
         gradientDescent(xs, y, 0)
         assert false
     }
 
     @Test(expected = IllegalArgumentException)
     void requiresMatchingParamSize() {
-        double[][] xs = [[1] as double[], [2] as double[], [3] as double[], [4] as double[]] as double[][]
-        double[] y = [2, 4, 5, 8] as double[]
         gradientDescent(xs, y, 1)
         assert false
     }
 
     @Test
     void simple1_9Line() {
-        double[][] xs = [[1] as double[], [2] as double[], [3] as double[], [4] as double[]] as double[][]
-        double[] y = [2, 4, 5, 8] as double[]
         double[] theta = gradientDescent(xs, y, 2)
         assert TestHelp.doubleEquals(theta, [0, 1.9] as double[])
     }