[mlpack-git] master: Refactor tests for changed NBC API. (0ae0b1a)
gitdub at big.cc.gt.atl.ga.us
gitdub at big.cc.gt.atl.ga.us
Tue Sep 29 09:33:41 EDT 2015
Repository : https://github.com/mlpack/mlpack
On branch : master
Link : https://github.com/mlpack/mlpack/compare/cbeb3ea17262b7c5115247dc217e316c529249b7...f85a9b22f3ce56143943a2488c05c2810d6b2bf3
>---------------------------------------------------------------
commit 0ae0b1a9a433aa548f68e91c984e39a5216c4fa7
Author: Ryan Curtin <ryan at ratml.org>
Date: Fri Sep 25 22:53:31 2015 +0000
Refactor tests for changed NBC API.
No new tests for Train() or empty constructor yet.
>---------------------------------------------------------------
0ae0b1a9a433aa548f68e91c984e39a5216c4fa7
src/mlpack/tests/nbc_test.cpp | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/mlpack/tests/nbc_test.cpp b/src/mlpack/tests/nbc_test.cpp
index a8b6ad7..4fdec14 100644
--- a/src/mlpack/tests/nbc_test.cpp
+++ b/src/mlpack/tests/nbc_test.cpp
@@ -27,7 +27,7 @@ BOOST_AUTO_TEST_CASE(NaiveBayesClassifierTest)
data::Load(trainResultFilename, trainRes, true);
// Get the labels out.
- arma::Col<size_t> labels(trainData.n_cols);
+ arma::Row<size_t> labels(trainData.n_cols);
for (size_t i = 0; i < trainData.n_cols; ++i)
labels[i] = trainData(trainData.n_rows - 1, i);
trainData.shed_row(trainData.n_rows - 1);
@@ -55,7 +55,7 @@ BOOST_AUTO_TEST_CASE(NaiveBayesClassifierTest)
arma::mat testData;
arma::Mat<size_t> testRes;
- arma::Col<size_t> calcVec;
+ arma::Row<size_t> calcVec;
data::Load(testFilename, testData, true);
data::Load(testResultFilename, testRes, true);
@@ -82,7 +82,7 @@ BOOST_AUTO_TEST_CASE(NaiveBayesClassifierIncrementalTest)
data::Load(trainResultFilename, trainRes, true);
// Get the labels out.
- arma::Col<size_t> labels(trainData.n_cols);
+ arma::Row<size_t> labels(trainData.n_cols);
for (size_t i = 0; i < trainData.n_cols; ++i)
labels[i] = trainData(trainData.n_rows - 1, i);
trainData.shed_row(trainData.n_rows - 1);
@@ -110,7 +110,7 @@ BOOST_AUTO_TEST_CASE(NaiveBayesClassifierIncrementalTest)
arma::mat testData;
arma::Mat<size_t> testRes;
- arma::Col<size_t> calcVec;
+ arma::Row<size_t> calcVec;
data::Load(testFilename, testData, true);
data::Load(testResultFilename, testRes, true);
More information about the mlpack-git
mailing list