[mlpack-git] master: Add test for CoverTreeDTNNKMeans. (2c4b63b)

gitdub at big.cc.gt.atl.ga.us gitdub at big.cc.gt.atl.ga.us
Thu Mar 5 22:01:49 EST 2015


Repository : https://github.com/mlpack/mlpack

On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/904762495c039e345beba14c1142fd719b3bd50e...f94823c800ad6f7266995c700b1b630d5ffdcf40

>---------------------------------------------------------------

commit 2c4b63b3f03a7dac4c9a24988535ba43e6b94ff0
Author: Ryan Curtin <ryan at ratml.org>
Date:   Tue Oct 14 06:15:25 2014 +0000

    Add test for CoverTreeDTNNKMeans.


>---------------------------------------------------------------

2c4b63b3f03a7dac4c9a24988535ba43e6b94ff0
 src/mlpack/tests/kmeans_test.cpp | 36 +++++++++++++++++++++++++++++++++++-
 1 file changed, 35 insertions(+), 1 deletion(-)

diff --git a/src/mlpack/tests/kmeans_test.cpp b/src/mlpack/tests/kmeans_test.cpp
index 0b22195..ef8aa2c 100644
--- a/src/mlpack/tests/kmeans_test.cpp
+++ b/src/mlpack/tests/kmeans_test.cpp
@@ -12,6 +12,8 @@
 #include <mlpack/methods/kmeans/pelleg_moore_kmeans.hpp>
 #include <mlpack/methods/kmeans/dtnn_kmeans.hpp>
 
+#include <mlpack/core/tree/cover_tree/cover_tree.hpp>
+
 #include <boost/test/unit_test.hpp>
 #include "old_boost_test_definitions.hpp"
 
@@ -618,7 +620,39 @@ BOOST_AUTO_TEST_CASE(DTNNTest)
     for (size_t i = 0; i < dataset.n_cols; ++i)
       BOOST_REQUIRE_EQUAL(assignments[i], dtnnAssignments[i]);
 
-    for (size_t i = 0; i < centroids.n_cols; ++i)
+    for (size_t i = 0; i < centroids.n_elem; ++i)
+      BOOST_REQUIRE_CLOSE(naiveCentroids[i], dtnnCentroids[i], 1e-5);
+  }
+}
+
+BOOST_AUTO_TEST_CASE(DTNNCoverTreeTest)
+{
+  const size_t trials = 5;
+
+  for (size_t t = 0; t < trials; ++t)
+  {
+    arma::mat dataset(10, 1000);
+    dataset.randu();
+
+    const size_t k = 5 * (t + 1);
+    arma::mat centroids(10, k);
+    centroids.randu();
+
+    arma::mat naiveCentroids(centroids);
+    KMeans<> km;
+    arma::Col<size_t> assignments;
+    km.Cluster(dataset, k, assignments, naiveCentroids, false, true);
+
+    KMeans<metric::EuclideanDistance, RandomPartition, MaxVarianceNewCluster,
+        CoverTreeDTNNKMeans> dtnn;
+    arma::Col<size_t> dtnnAssignments;
+    arma::mat dtnnCentroids(centroids);
+    dtnn.Cluster(dataset, k, dtnnAssignments, dtnnCentroids, false, true);
+
+    for (size_t i = 0; i < dataset.n_cols; ++i)
+      BOOST_REQUIRE_EQUAL(assignments[i], dtnnAssignments[i]);
+
+    for (size_t i = 0; i < centroids.n_elem; ++i)
       BOOST_REQUIRE_CLOSE(naiveCentroids[i], dtnnCentroids[i], 1e-5);
   }
 }



More information about the mlpack-git mailing list