[mlpack-svn] r16975 - mlpack/trunk/src/mlpack/methods/naive_bayes

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

Author: rcurtin
Date: Wed Aug  6 11:01:45 2014
New Revision: 16975

Log:
Oops, I fixed this backwards.  The actual error is the output.


Modified:
   mlpack/trunk/src/mlpack/methods/naive_bayes/nbc_main.cpp

Modified: mlpack/trunk/src/mlpack/methods/naive_bayes/nbc_main.cpp
==============================================================================
--- mlpack/trunk/src/mlpack/methods/naive_bayes/nbc_main.cpp	(original)
+++ mlpack/trunk/src/mlpack/methods/naive_bayes/nbc_main.cpp	Wed Aug  6 11:01:45 2014
@@ -81,9 +81,9 @@
   mat testingData;
   data::Load(testingDataFilename, testingData, true);
 
-  if (testingData.n_rows != (trainingData.n_rows - 1))
+  if (testingData.n_rows != trainingData.n_rows)
     Log::Fatal << "Test data dimensionality (" << testingData.n_rows << ") "
-        << "must be the same as training data (" << trainingData.n_rows - 1
+        << "must be the same as training data (" << trainingData.n_rows
         << ")!" << std::endl;
 
   const bool incrementalVariance = CLI::HasParam("incremental_variance");