From cbe14aa8f7d8bc36ca44d02db68e1ade75a245be Mon Sep 17 00:00:00 2001 From: daivik Date: Tue, 13 Feb 2018 12:17:00 +0000 Subject: [PATCH] Minor changes --- src/mlpack/methods/hmm/hmm_train_main.cpp | 7 ++----- src/mlpack/tests/CMakeLists.txt | 1 + .../{hmm_training_tests.cpp => hmm_train_test.cpp} | 14 +++++++------- 3 files changed, 10 insertions(+), 12 deletions(-) rename src/mlpack/tests/main_tests/{hmm_training_tests.cpp => hmm_train_test.cpp} (96%) diff --git a/src/mlpack/methods/hmm/hmm_train_main.cpp b/src/mlpack/methods/hmm/hmm_train_main.cpp index 9357e8e01c..67eca0af21 100644 --- a/src/mlpack/methods/hmm/hmm_train_main.cpp +++ b/src/mlpack/methods/hmm/hmm_train_main.cpp @@ -379,11 +379,8 @@ static void mlpackMain() "unknown HMM type"); } - if (CLI::HasParam("tolerance")) - { - RequireParamValue("tolerance", [](int x) { return x >= 0; }, true, - "tolerance must be non-negative"); - } + RequireParamValue("tolerance", [](double x) { return x >= 0; }, true, + "tolerance must be non-negative"); // Load the input data. vector trainSeq; diff --git a/src/mlpack/tests/CMakeLists.txt b/src/mlpack/tests/CMakeLists.txt index 86251f6e69..597330d7c3 100644 --- a/src/mlpack/tests/CMakeLists.txt +++ b/src/mlpack/tests/CMakeLists.txt @@ -145,6 +145,7 @@ add_executable(mlpack_test main_tests/sparse_coding_test.cpp main_tests/hoeffding_tree_test.cpp main_tests/hmm_training_tests.cpp + main_tests/hmm_train_test.cpp ) # Link dependencies of test executable. diff --git a/src/mlpack/tests/main_tests/hmm_training_tests.cpp b/src/mlpack/tests/main_tests/hmm_train_test.cpp similarity index 96% rename from src/mlpack/tests/main_tests/hmm_training_tests.cpp rename to src/mlpack/tests/main_tests/hmm_train_test.cpp index e79aa07570..308234e8e5 100644 --- a/src/mlpack/tests/main_tests/hmm_training_tests.cpp +++ b/src/mlpack/tests/main_tests/hmm_train_test.cpp @@ -43,7 +43,7 @@ struct HMMTrainMainTestFixture BOOST_FIXTURE_TEST_SUITE(HMMTrainMainTest, HMMTrainMainTestFixture); -inline void fileExists(std::string fileName) +inline void FileExists(std::string fileName) { ifstream ifp(fileName); if (!ifp.good()) @@ -58,7 +58,7 @@ BOOST_AUTO_TEST_CASE(HMMTrainStatesTest) int states = -3; // Invalid! std::string hmmType = "discrete"; - fileExists(inputFileName); + FileExists(inputFileName); SetInputParam("input_file", std::move(inputFileName)); SetInputParam("states", states); SetInputParam("type", std::move(hmmType)); @@ -76,7 +76,7 @@ BOOST_AUTO_TEST_CASE(HMMTrainToleranceNonNegative) std::string hmmType = "gaussian"; double tol = - 100; // Invalid - fileExists(inputFileName); + FileExists(inputFileName); SetInputParam("input_file", std::move(inputFileName)); SetInputParam("states", states); SetInputParam("type", std::move(hmmType)); @@ -95,7 +95,7 @@ BOOST_AUTO_TEST_CASE(HMMTrainTypeTest) int states = 3; std::string hmmType = "some-not-supported-possibly-non-type"; - fileExists(inputFileName); + FileExists(inputFileName); SetInputParam("input_file", std::move(inputFileName)); SetInputParam("states", states); SetInputParam("type", std::move(hmmType)); @@ -113,7 +113,7 @@ BOOST_AUTO_TEST_CASE(HMMTrainGaussianTest) std::string hmmType = "gmm"; int gaussians = -2; - fileExists(inputFileName); + FileExists(inputFileName); SetInputParam("input_file", std::move(inputFileName)); SetInputParam("states", states); SetInputParam("type", std::move(hmmType)); @@ -132,8 +132,8 @@ BOOST_AUTO_TEST_CASE(HMMTrainReuseModelTest) std::string hmmType = "discrete"; int states = 3; - fileExists(inputObsFileName); - fileExists(inputLabFileName); + FileExists(inputObsFileName); + FileExists(inputLabFileName); // Make sure that the size of the // training seq, and training labels is same arma::mat trainObs, trainLab;