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

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

Author: rcurtin
Date: Mon Apr 14 15:26:36 2014
New Revision: 16424

Log:
Get the number of classes correctly.


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	Mon Apr 14 15:26:36 2014
@@ -80,12 +80,9 @@
         << "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)) + 1;
-
   // Create and train the classifier.
   Timer::Start("training");
-  NaiveBayesClassifier<> nbc(trainingData, labels, classes);
+  NaiveBayesClassifier<> nbc(trainingData, labels, mappings.n_elem);
   Timer::Stop("training");
 
   // Time the running of the Naive Bayes Classifier.