Author: tommaso
Date: Sat Nov 3 13:42:10 2012
New Revision: 1405331
URL: http://svn.apache.org/viewvc?rev=1405331&view=rev
Log:
[HAMA-666] - adding unit test for linear regression
Added:
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/LinearRegressionModelTest.java
(with props)
Added:
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/LinearRegressionModelTest.java
URL:
http://svn.apache.org/viewvc/hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/LinearRegressionModelTest.java?rev=1405331&view=auto
==============================================================================
---
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/LinearRegressionModelTest.java
(added)
+++
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/LinearRegressionModelTest.java
Sat Nov 3 13:42:10 2012
@@ -0,0 +1,23 @@
+package org.apache.hama.ml.regression;
+
+import org.apache.hama.ml.math.DenseDoubleVector;
+import org.apache.hama.ml.math.DoubleVector;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+
+/**
+ * Testcase for {@link LinearRegressionModel}
+ */
+public class LinearRegressionModelTest {
+
+ @Test
+ public void testCorrectCalculation() throws Exception {
+ LinearRegressionModel linearRegressionModel = new LinearRegressionModel();
+ DoubleVector x = new DenseDoubleVector(new double[]{2, 3, 4});
+ double y = 1;
+ DoubleVector theta = new DenseDoubleVector(new double[]{1, 1, 1});
+ Double cost = linearRegressionModel.calculateCostForItem(x, y, 2, theta);
+ assertEquals("wrong cost calculation for linear regression",
Double.valueOf(16d), cost);
+ }
+}
Propchange:
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/LinearRegressionModelTest.java
------------------------------------------------------------------------------
svn:eol-style = native
