[mlpack-svn] r17301 - mlpack/trunk/src/mlpack/tests
fastlab-svn at coffeetalk-1.cc.gatech.edu
fastlab-svn at coffeetalk-1.cc.gatech.edu
Wed Nov 5 14:37:28 EST 2014
Author: rcurtin
Date: Wed Nov 5 14:37:27 2014
New Revision: 17301
Log:
Add a simple test for DualTreeKMeans.
Modified:
mlpack/trunk/src/mlpack/tests/kmeans_test.cpp
Modified: mlpack/trunk/src/mlpack/tests/kmeans_test.cpp
==============================================================================
--- mlpack/trunk/src/mlpack/tests/kmeans_test.cpp (original)
+++ mlpack/trunk/src/mlpack/tests/kmeans_test.cpp Wed Nov 5 14:37:27 2014
@@ -11,6 +11,7 @@
#include <mlpack/methods/kmeans/hamerly_kmeans.hpp>
#include <mlpack/methods/kmeans/pelleg_moore_kmeans.hpp>
#include <mlpack/methods/kmeans/dtnn_kmeans.hpp>
+#include <mlpack/methods/kmeans/dual_tree_kmeans.hpp>
#include <mlpack/core/tree/cover_tree/cover_tree.hpp>
@@ -657,4 +658,36 @@
}
}
+BOOST_AUTO_TEST_CASE(DualTreeKMeansTest)
+{
+ 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,
+ DefaultDualTreeKMeans> 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);
+ }
+}
+
BOOST_AUTO_TEST_SUITE_END();
More information about the mlpack-svn
mailing list