From 148bfca48cccba1647ffcfacd03e578a493b7265 Mon Sep 17 00:00:00 2001 From: Ryan Curtin Date: Thu, 7 May 2015 15:01:05 -0400 Subject: [PATCH] Use MidpointSplit by default. --- src/mlpack/core/tree/CMakeLists.txt | 2 ++ src/mlpack/core/tree/binary_space_tree.hpp | 2 ++ src/mlpack/core/tree/binary_space_tree/binary_space_tree.hpp | 4 ++-- 3 files changed, 6 insertions(+), 2 deletions(-) diff --git a/src/mlpack/core/tree/CMakeLists.txt b/src/mlpack/core/tree/CMakeLists.txt index 7000b3ff4f..82fd8d90e8 100644 --- a/src/mlpack/core/tree/CMakeLists.txt +++ b/src/mlpack/core/tree/CMakeLists.txt @@ -11,6 +11,8 @@ set(SOURCES binary_space_tree/dual_tree_traverser_impl.hpp binary_space_tree/mean_split.hpp binary_space_tree/mean_split_impl.hpp + binary_space_tree/midpoint_split.hpp + binary_space_tree/midpoint_split_impl.hpp binary_space_tree/single_tree_traverser.hpp binary_space_tree/single_tree_traverser_impl.hpp binary_space_tree/traits.hpp diff --git a/src/mlpack/core/tree/binary_space_tree.hpp b/src/mlpack/core/tree/binary_space_tree.hpp index 91e3eaf009..374c3358c6 100644 --- a/src/mlpack/core/tree/binary_space_tree.hpp +++ b/src/mlpack/core/tree/binary_space_tree.hpp @@ -16,5 +16,7 @@ #include "binary_space_tree/breadth_first_dual_tree_traverser.hpp" #include "binary_space_tree/breadth_first_dual_tree_traverser_impl.hpp" #include "binary_space_tree/traits.hpp" +#include "binary_space_tree/midpoint_split.hpp" +#include "binary_space_tree/mean_split.hpp" #endif diff --git a/src/mlpack/core/tree/binary_space_tree/binary_space_tree.hpp b/src/mlpack/core/tree/binary_space_tree/binary_space_tree.hpp index cbd6da75fa..caab084cbe 100644 --- a/src/mlpack/core/tree/binary_space_tree/binary_space_tree.hpp +++ b/src/mlpack/core/tree/binary_space_tree/binary_space_tree.hpp @@ -7,7 +7,7 @@ #define __MLPACK_CORE_TREE_BINARY_SPACE_TREE_BINARY_SPACE_TREE_HPP #include -#include "mean_split.hpp" +#include "midpoint_split.hpp" #include "../statistic.hpp" @@ -40,7 +40,7 @@ namespace tree /** Trees and tree-building procedures. */ { template > + typename SplitType = MidpointSplit> class BinarySpaceTree { private: