[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