[mlpack-git] master: refine test cases (a4d925c)
gitdub at big.cc.gt.atl.ga.us
gitdub at big.cc.gt.atl.ga.us
Wed Sep 30 09:27:21 EDT 2015
Repository : https://github.com/mlpack/mlpack
On branch : master
Link : https://github.com/mlpack/mlpack/compare/8a8b708650f72c2aecbd9b4a12c8b16b4e0a3508...dc2c5c68dc4bfcdd2075b1a0fd2d641fce651669
>---------------------------------------------------------------
commit a4d925c500a90ce01c379ab4240c89666fa528e2
Author: stereomatchingkiss <stereomatchingkiss at gmail.com>
Date: Tue Sep 29 13:03:56 2015 +0800
refine test cases
>---------------------------------------------------------------
a4d925c500a90ce01c379ab4240c89666fa528e2
src/mlpack/tests/serialization_test.cpp | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/mlpack/tests/serialization_test.cpp b/src/mlpack/tests/serialization_test.cpp
index ac91561..18b8e81 100644
--- a/src/mlpack/tests/serialization_test.cpp
+++ b/src/mlpack/tests/serialization_test.cpp
@@ -805,13 +805,13 @@ BOOST_AUTO_TEST_CASE(SoftmaxRegressionTest)
using regression::SoftmaxRegression;
arma::mat dataset = arma::randu<arma::mat>(5, 1000);
- arma::vec labels(1000);
+ arma::Row<size_t> labels(1000);
for (size_t i = 0; i < 500; ++i)
- labels[i] = 0.0;
+ labels[i] = 0;
for (size_t i = 500; i < 1000; ++i)
- labels[i] = 1.0;
+ labels[i] = 1;
- SoftmaxRegression<> sr(dataset, labels, dataset.n_rows, 2);
+ SoftmaxRegression<> sr(dataset, labels, 2);
SoftmaxRegression<> srXml(dataset.n_rows, 2);
SoftmaxRegression<> srText(dataset.n_rows, 2);
More information about the mlpack-git
mailing list