Reduce dataset sizes and number of iterations to accelerate tests.

This commit is contained in:
Ryan Curtin
2017-08-08 23:21:48 -04:00
parent 54302d3272
commit 68f22bc1ba
3 changed files with 4 additions and 4 deletions
+2 -2
View File
@@ -608,7 +608,7 @@ BOOST_AUTO_TEST_CASE(DTNNTest)
for (size_t t = 0; t < trials; ++t)
{
arma::mat dataset(10, 1000);
arma::mat dataset(10, 300);
dataset.randu();
const size_t k = 5 * (t + 1);
@@ -640,7 +640,7 @@ BOOST_AUTO_TEST_CASE(DTNNCoverTreeTest)
for (size_t t = 0; t < trials; ++t)
{
arma::mat dataset(10, 1000);
arma::mat dataset(10, 300);
dataset.randu();
const size_t k = 5;
+1 -1
View File
@@ -121,7 +121,7 @@ BOOST_AUTO_TEST_CASE(LogisticRegressionTest)
// Now run mini-batch SGD with a couple of batch sizes.
for (size_t batchSize = 5; batchSize < 50; batchSize += 5)
{
MiniBatchSGD mbsgd(batchSize);
MiniBatchSGD mbsgd(batchSize, 0.01, 10000, 1e-3);
LogisticRegression<> lr(shuffledData, shuffledResponses, mbsgd, 0.5);
// Ensure that the error is close to zero.
+1 -1
View File
@@ -251,7 +251,7 @@ void CheckDistance(TreeType& tree, TreeType* node = NULL)
BOOST_AUTO_TEST_CASE(UBTreeDistanceTest)
{
typedef UBTree<EuclideanDistance, EmptyStatistic, arma::mat> TreeType;
arma::mat dataset(8, 1000);
arma::mat dataset(8, 200);
dataset.randu();