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

Merge pull request #2751 from Abilityguy/FailMessage

Browse Source 
Added FAIL messages to svd_batch_test and argument fix for callback_test.
This commit is contained in:
jeffin sam
2020-12-04 23:54:35 +05:30
committed by GitHub
parent 0b924c09ea 6cad509c2f
commit 4c180e3fdf
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/tests/callback_test.cpp
+4 -4
View File
@@ -42,9 +42,9 @@ TEST_CASE("FFNCallbackTest", "[CallbackTest]")
arma::mat data;
arma::mat labels;
if (!data::Load("lab1.csv", data, true))
if (!data::Load("lab1.csv", data))
FAIL("Cannot load test dataset lab1.csv!");
if (!data::Load("lab3.csv", labels, true))
if (!data::Load("lab3.csv", labels))
FAIL("Cannot load test dataset lab3.csv!");
FFN<MeanSquaredError<>, RandomInitialization> model;
@@ -68,9 +68,9 @@ TEST_CASE("FFNWithOptimizerCallbackTest", "[CallbackTest]")
arma::mat data;
arma::mat labels;
if (!data::Load("lab1.csv", data, true))
if (!data::Load("lab1.csv", data))
FAIL("Cannot load test dataset lab1.csv!");
if (!data::Load("lab3.csv", labels, true))
if (!data::Load("lab3.csv", labels))
FAIL("Cannot load test dataset lab3.csv!");
FFN<MeanSquaredError<>, RandomInitialization> model;
src/mlpack/tests/svd_batch_test.cpp
+4 -2
View File
@@ -70,7 +70,8 @@ class SpecificRandomInitialization
TEST_CASE("SVDBatchMomentumTest", "[SVDBatchTest]")
{
mat dataset;
data::Load("GroupLensSmall.csv", dataset);
if (!data::Load("GroupLensSmall.csv", dataset))
FAIL("Cannot load dataset GroupLensSmall.csv!");
// Generate list of locations for batch insert constructor for sparse
// matrices.
@@ -117,7 +118,8 @@ TEST_CASE("SVDBatchMomentumTest", "[SVDBatchTest]")
TEST_CASE("SVDBatchRegularizationTest", "[SVDBatchTest]")
{
mat dataset;
data::Load("GroupLensSmall.csv", dataset);
if (!data::Load("GroupLensSmall.csv", dataset))
FAIL("Cannot load dataset GroupLensSmall.csv!");
// Generate list of locations for batch insert constructor for sparse
// matrices.