From 4ee34f4237a22cd660bc13e1deafbb9c2d3707cf Mon Sep 17 00:00:00 2001 From: Marcus Edel Date: Sat, 17 May 2014 13:59:14 +0000 Subject: [PATCH] Save the correct labels; thanks Anand! --- src/mlpack/methods/naive_bayes/nbc_main.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/mlpack/methods/naive_bayes/nbc_main.cpp b/src/mlpack/methods/naive_bayes/nbc_main.cpp index 391886c020..0bfa4f9b9d 100644 --- a/src/mlpack/methods/naive_bayes/nbc_main.cpp +++ b/src/mlpack/methods/naive_bayes/nbc_main.cpp @@ -106,5 +106,5 @@ int main(int argc, char* argv[]) // Output results. Don't transpose: one result per line. const string outputFilename = CLI::GetParam("output"); - data::Save(outputFilename, results, true, false); + data::Save(outputFilename, rawResults, true, false); }