[mlpack] 92/207: Adapt decision tree test to new DiscreteDistribution API.

Barak A. Pearlmutter barak+git at pearlmutter.net
Thu Mar 23 17:53:43 UTC 2017


This is an automated email from the git hooks/post-receive script.

bap pushed a commit to branch master
in repository mlpack.

commit d27edce0fa9e3ac8fe727aa9608ac05848284dcb
Author: Ryan Curtin <ryan at ratml.org>
Date:   Sat Feb 25 15:09:58 2017 -0500

    Adapt decision tree test to new DiscreteDistribution API.
---
 src/mlpack/tests/decision_tree_test.cpp | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/src/mlpack/tests/decision_tree_test.cpp b/src/mlpack/tests/decision_tree_test.cpp
index 02f8733..1caa34b 100644
--- a/src/mlpack/tests/decision_tree_test.cpp
+++ b/src/mlpack/tests/decision_tree_test.cpp
@@ -461,19 +461,25 @@ BOOST_AUTO_TEST_CASE(SimpleGeneralizationTest)
  */
 BOOST_AUTO_TEST_CASE(CategoricalBuildTest)
 {
-  math::RandomSeed(std::time(NULL));
-
   // We'll build a spiral dataset plus two noisy categorical features.  We need
   // to build the distributions for the categorical features (they'll be
   // discrete distributions).
   DiscreteDistribution c1[5];
   // The distribution will be automatically normalized.
   for (size_t i = 0; i < 5; ++i)
-    c1[i] = DiscreteDistribution(arma::vec(4, arma::fill::randu));
+  {
+    std::vector<arma::vec> probs;
+    probs.push_back(arma::vec(4, arma::fill::randu));
+    c1[i] = DiscreteDistribution(probs);
+  }
 
   DiscreteDistribution c2[5];
   for (size_t i = 0; i < 5; ++i)
-    c2[i] = DiscreteDistribution(arma::vec(2, arma::fill::randu));
+  {
+    std::vector<arma::vec> probs;
+    probs.push_back(arma::vec(2, arma::fill::randu));
+    c2[i] = DiscreteDistribution(probs);
+  }
 
   arma::mat spiralDataset(4, 10000);
   arma::Row<size_t> labels(10000);

-- 
Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-science/packages/mlpack.git



More information about the debian-science-commits mailing list