[mlpack] 72/149: Add DualTreeKMeans as an option to the kmeans program (and also DTNN with cover trees).
Barak A. Pearlmutter
barak+git at pearlmutter.net
Sat May 2 09:11:10 UTC 2015
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
More information about the debian-science-commits
mailing list