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

fastlab-svn at coffeetalk-1.cc.gatech.edu fastlab-svn at coffeetalk-1.cc.gatech.edu
Wed Dec 14 17:13:14 EST 2011


Author: rcurtin
Date: 2011-12-14 17:13:14 -0500 (Wed, 14 Dec 2011)
New Revision: 10809

Modified:
   mlpack/trunk/src/mlpack/methods/naive_bayes/nbc_main.cpp
Log:
Fix bug and better warnings.


Modified: mlpack/trunk/src/mlpack/methods/naive_bayes/nbc_main.cpp
===================================================================
--- mlpack/trunk/src/mlpack/methods/naive_bayes/nbc_main.cpp	2011-12-14 22:08:08 UTC (rev 10808)
+++ mlpack/trunk/src/mlpack/methods/naive_bayes/nbc_main.cpp	2011-12-14 22:13:14 UTC (rev 10809)
@@ -63,8 +63,13 @@
   mat testingData;
   data::Load(testingDataFilename.c_str(), testingData, true);
 
+  if (testingData.n_rows != trainingData.n_rows - 1)
+    Log::Fatal << "Test data dimensionality (" << testingData.n_rows << ") "
+        << "must be the same as training data (" << trainingData.n_rows - 1
+        << ")!" << std::endl;
+
   // Calculate number of classes.
-  size_t classes = (size_t) max(trainingData.row(trainingData.n_rows - 1));
+  size_t classes = (size_t) max(trainingData.row(trainingData.n_rows - 1)) + 1;
 
   // Create and train the classifier.
   Timer::Start("training");




More information about the mlpack-svn mailing list