diff --git a/src/mlpack/methods/kmeans/dtnn_kmeans_impl.hpp b/src/mlpack/methods/kmeans/dtnn_kmeans_impl.hpp index e6747e5f98..1f4854504d 100644 --- a/src/mlpack/methods/kmeans/dtnn_kmeans_impl.hpp +++ b/src/mlpack/methods/kmeans/dtnn_kmeans_impl.hpp @@ -29,7 +29,7 @@ TreeType* BuildTree( { // This is a hack. I know this will be BinarySpaceTree, so force a leaf size // of two. - return new TreeType(dataset, oldFromNew, 1); + return new TreeType(dataset, oldFromNew, 4); } //! Call the tree constructor that does not do mapping.