From d9d69fcc7f402473465557b565dd2bb105bb6ad2 Mon Sep 17 00:00:00 2001 From: Ryan Curtin Date: Mon, 19 Oct 2015 14:17:40 -0400 Subject: [PATCH] Make some test sets smaller for speed. --- src/mlpack/tests/allknn_test.cpp | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/mlpack/tests/allknn_test.cpp b/src/mlpack/tests/allknn_test.cpp index cf22d5e68a..1ef3620053 100644 --- a/src/mlpack/tests/allknn_test.cpp +++ b/src/mlpack/tests/allknn_test.cpp @@ -765,7 +765,7 @@ BOOST_AUTO_TEST_CASE(DualCoverTreeTest) BOOST_AUTO_TEST_CASE(SingleBallTreeTest) { arma::mat data; - data.randu(75, 1000); // 75 dimensional, 1000 points. + data.randu(50, 300); // 50 dimensional, 300 points. typedef BallTree, arma::mat> TreeType; @@ -832,9 +832,9 @@ BOOST_AUTO_TEST_CASE(SparseAllkNNKDTreeTest) // 70, the probability of all 70 dimensions being zero is 0.8^70 = 1.65e-7 in // the reference set and 0.9^70 = 6.27e-4 in the query set. arma::sp_mat queryDataset; - queryDataset.sprandu(70, 500, 0.2); + queryDataset.sprandu(70, 200, 0.2); arma::sp_mat referenceDataset; - referenceDataset.sprandu(70, 800, 0.1); + referenceDataset.sprandu(70, 500, 0.1); arma::mat denseQuery(queryDataset); arma::mat denseReference(referenceDataset); @@ -907,8 +907,8 @@ BOOST_AUTO_TEST_CASE(KNNModelTest) // results. typedef NSModel KNNModel; - arma::mat queryData = arma::randu(10, 100); - arma::mat referenceData = arma::randu(10, 500); + arma::mat queryData = arma::randu(10, 50); + arma::mat referenceData = arma::randu(10, 200); // Build all the possible models. KNNModel models[8]; @@ -967,7 +967,7 @@ BOOST_AUTO_TEST_CASE(KNNModelMonochromaticTest) // results, in the case where the reference set is the same as the query set. typedef NSModel KNNModel; - arma::mat referenceData = arma::randu(10, 500); + arma::mat referenceData = arma::randu(10, 200); // Build all the possible models. KNNModel models[8];