apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

SoftmaxRegression Fix

Browse Source
This commit is contained in:
manish7294
2018-01-18 19:32:01 +05:30
parent 8a09b47d9a
commit f10866226f
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/tests/main_tests/softmax_regression_test.cpp
+10 -10
View File
@@ -348,18 +348,18 @@ BOOST_AUTO_TEST_CASE(SoftmaxRegressionDiffLambdaTest)
// Check that initial parameters and final parameters matrix
// using saved model are different.
bool flag = true;
bool flag = false;
bool* flagPtr = &flag;
for (size_t i = 0; i < modelParam.n_elem; ++i)
{
if ((int) modelParam[i] * 1e+6 == (int) CLI::GetParam<SoftmaxRegression>
("output_model").Parameters()[i] * 1e+6)
if ((int) (modelParam[i] * 1e+6) != (int) (CLI::GetParam<SoftmaxRegression>
("output_model").Parameters()[i] * 1e+6))
{
*flagPtr = false;
*flagPtr = true;
break;
}
}
BOOST_REQUIRE_EQUAL(flag, false);
BOOST_REQUIRE_EQUAL(flag, true);
}
/**
@@ -427,18 +427,18 @@ BOOST_AUTO_TEST_CASE(SoftmaxRegressionDiffMaxItrTest)
// Check that initial parameters and final parameters matrix
// using saved model are different.
bool flag = true;
bool flag = false;
bool* flagPtr = &flag;
for (size_t i = 0; i < modelParam.n_elem; ++i)
{
if ((int) modelParam[i] * 1e+6 == (int) CLI::GetParam<SoftmaxRegression>
("output_model").Parameters()[i] * 1e+6)
if ((int) (modelParam[i] * 1e+6) != (int) (CLI::GetParam<SoftmaxRegression>
("output_model").Parameters()[i] * 1e+6))
{
*flagPtr = false;
*flagPtr = true;
break;
}
}
BOOST_REQUIRE_EQUAL(flag, false);
BOOST_REQUIRE_EQUAL(flag, true);
}
/**