[mlpack-git] master: Add DualTreeKMeans as an option to the kmeans program (and also DTNN with cover trees). (bdc8824)

[gitdub at big.cc.gt.atl.ga.us]

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

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

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

commit bdc8824f4de0a1d83747d60f1b5abfa598ae2d4c
Author: Ryan Curtin <ryan at ratml.org>
Date:   Wed Nov 5 19:37:18 2014 +0000

    Add DualTreeKMeans as an option to the kmeans program (and also DTNN with cover
    trees).


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

bdc8824f4de0a1d83747d60f1b5abfa598ae2d4c
 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