[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