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

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

Author: rcurtin
Date: 2011-12-14 01:43:10 -0500 (Wed, 14 Dec 2011)
New Revision: 10759

Modified:
   mlpack/trunk/src/mlpack/methods/naive_bayes/nbc_main.cpp
Log:
Oops, I need a <>.


Modified: mlpack/trunk/src/mlpack/methods/naive_bayes/nbc_main.cpp
===================================================================
--- mlpack/trunk/src/mlpack/methods/naive_bayes/nbc_main.cpp	2011-12-13 22:28:33 UTC (rev 10758)
+++ mlpack/trunk/src/mlpack/methods/naive_bayes/nbc_main.cpp	2011-12-14 06:43:10 UTC (rev 10759)
@@ -59,12 +59,11 @@
 
   // Create and train the classifier.
   Timers::StartTimer("training");
-  NaiveBayesClassifier nbc = NaiveBayesClassifier(training_data,
-      number_of_classes_);
+  NaiveBayesClassifier<> nbc(training_data, number_of_classes_);
   Timers::StopTimer("training");
 
   // Timing the running of the Naive Bayes Classifier.
-  arma::vec results;
+  arma::Col<size_t> results;
   Timers::StartTimer("testing");
   nbc.Classify(testing_data, results);
   Timers::StopTimer("testing");