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); }