[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