[mlpack-svn] r16706 - mlpack/trunk/src/mlpack/tests

[fastlab-svn at coffeetalk-1.cc.gatech.edu]

Author: saxena.udit
Date: Tue Jun 24 14:17:55 2014
New Revision: 16706

Log:
Decision Stump test fixed

Modified:
   mlpack/trunk/src/mlpack/tests/decision_stump_test.cpp

Modified: mlpack/trunk/src/mlpack/tests/decision_stump_test.cpp
==============================================================================
--- mlpack/trunk/src/mlpack/tests/decision_stump_test.cpp	(original)
+++ mlpack/trunk/src/mlpack/tests/decision_stump_test.cpp	Tue Jun 24 14:17:55 2014
@@ -75,7 +75,11 @@
   Row<size_t> predictedLabels(testingData.n_cols);
   ds.Classify(testingData, predictedLabels);
 
-  data::Save(output, predictedLabels, true, true);
+  BOOST_CHECK_EQUAL(predictedLabels(0,0),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,1),1);
+  BOOST_CHECK_EQUAL(predictedLabels(0,2),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,3),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,4),1);
 }
 
 /**
@@ -103,7 +107,7 @@
   Row<size_t> predictedLabels(testingData.n_cols);
   ds.Classify(testingData, predictedLabels);
 
-  data::Save(output, predictedLabels, true, true);
+  BOOST_CHECK_EQUAL(predictedLabels(0,0),0);
 }
 
 /**
@@ -134,7 +138,10 @@
   Row<size_t> predictedLabels(testingData.n_cols);
   ds.Classify(testingData, predictedLabels);
 
-  data::Save(output, predictedLabels, true, true);
+  BOOST_CHECK_EQUAL(predictedLabels(0,0),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,1),1);
+  BOOST_CHECK_EQUAL(predictedLabels(0,2),2);
+  BOOST_CHECK_EQUAL(predictedLabels(0,3),3);
 }
 
 /**
@@ -167,7 +174,14 @@
   Row<size_t> predictedLabels(testingData.n_cols);
   ds.Classify(testingData, predictedLabels);
 
-  data::Save(output, predictedLabels, true, true);
+  BOOST_CHECK_EQUAL(predictedLabels(0,0),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,1),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,2),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,3),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,4),0);
+  BOOST_CHECK_EQUAL(predictedLabels(0,5),1);
+  BOOST_CHECK_EQUAL(predictedLabels(0,6),2);
+  BOOST_CHECK_EQUAL(predictedLabels(0,7),2);
 }
 
 BOOST_AUTO_TEST_SUITE_END();