[mlpack] 72/149: Add DualTreeKMeans as an option to the kmeans program (and also DTNN with cover trees).

[Barak A. Pearlmutter]

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

bap pushed a commit to branch svn-trunk
in repository mlpack.

commit 26ed8e1475d964686d7d3d90d9110f56c798a472
Author: rcurtin <rcurtin at 9d5b8971-822b-0410-80eb-d18c1038ef23>
Date:   Wed Nov 5 19:37:18 2014 +0000

    Add DualTreeKMeans as an option to the kmeans program (and also DTNN with cover
    trees).
    
    
    git-svn-id: http://svn.cc.gatech.edu/fastlab/mlpack/trunk@17300 9d5b8971-822b-0410-80eb-d18c1038ef23
---
 src/mlpack/methods/kmeans/kmeans_main.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/mlpack/methods/kmeans/kmeans_main.cpp b/src/mlpack/methods/kmeans/kmeans_main.cpp
index 039925c..331eea1 100644
--- a/src/mlpack/methods/kmeans/kmeans_main.cpp
+++ b/src/mlpack/methods/kmeans/kmeans_main.cpp
@@ -13,6 +13,7 @@
 #include "hamerly_kmeans.hpp"
 #include "pelleg_moore_kmeans.hpp"
 #include "dtnn_kmeans.hpp"
+#include "dual_tree_kmeans.hpp"
 
 using namespace mlpack;
 using namespace mlpack::kmeans;
@@ -154,6 +155,12 @@ void FindLloydStepType(const InitialPartitionPolicy& ipp)
   else if (algorithm == "dtnn")
     RunKMeans<InitialPartitionPolicy, EmptyClusterPolicy,
         DefaultDTNNKMeans>(ipp);
+  else if (algorithm == "dtnn-covertree")
+    RunKMeans<InitialPartitionPolicy, EmptyClusterPolicy,
+        CoverTreeDTNNKMeans>(ipp);
+  else if (algorithm == "dualtree")
+    RunKMeans<InitialPartitionPolicy, EmptyClusterPolicy,
+        DefaultDualTreeKMeans>(ipp);
   else if (algorithm == "naive")
     RunKMeans<InitialPartitionPolicy, EmptyClusterPolicy, NaiveKMeans>(ipp);
   else

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