[mlpack-git] master: Minor refactorings and fixes. (9e3550b)
gitdub at big.cc.gt.atl.ga.us
gitdub at big.cc.gt.atl.ga.us
Wed Jul 29 16:42:01 EDT 2015
Repository : https://github.com/mlpack/mlpack
On branch : master
Link : https://github.com/mlpack/mlpack/compare/f8ceffae0613b350f4d6bdd46c6c8633a40b4897...6ee21879488fe98612a4619b17f8b51e8da5215b
>---------------------------------------------------------------
commit 9e3550b8ca636ce1de183b6f9ea6b17f823f3db8
Author: ryan <ryan at ratml.org>
Date: Fri Jul 24 14:27:30 2015 -0400
Minor refactorings and fixes.
>---------------------------------------------------------------
9e3550b8ca636ce1de183b6f9ea6b17f823f3db8
.../core/tree/rectangle_tree/r_star_tree_split_impl.hpp | 2 +-
src/mlpack/core/tree/rectangle_tree/traits.hpp | 14 ++++++++------
2 files changed, 9 insertions(+), 7 deletions(-)
diff --git a/src/mlpack/core/tree/rectangle_tree/r_star_tree_split_impl.hpp b/src/mlpack/core/tree/rectangle_tree/r_star_tree_split_impl.hpp
index 8235d8f..1a26bc0 100644
--- a/src/mlpack/core/tree/rectangle_tree/r_star_tree_split_impl.hpp
+++ b/src/mlpack/core/tree/rectangle_tree/r_star_tree_split_impl.hpp
@@ -64,7 +64,7 @@ void RStarTreeSplit::SplitLeafNode(TreeType* tree, std::vector<bool>& relevels)
tree->Bound().Center(center); // Modifies centroid.
for (size_t i = 0; i < sorted.size(); i++)
{
- sorted[i].d = tree->Bound().Metric().Evaluate(center,
+ sorted[i].d = tree->Metric().Evaluate(center,
tree->LocalDataset().col(i));
sorted[i].n = i;
}
diff --git a/src/mlpack/core/tree/rectangle_tree/traits.hpp b/src/mlpack/core/tree/rectangle_tree/traits.hpp
index 0839c78..811d9ac 100644
--- a/src/mlpack/core/tree/rectangle_tree/traits.hpp
+++ b/src/mlpack/core/tree/rectangle_tree/traits.hpp
@@ -18,11 +18,13 @@ namespace tree {
* help write tree-independent (but still optimized) tree-based algorithms. See
* mlpack/core/tree/tree_traits.hpp for more information.
*/
-template<typename SplitType,
- typename DescentType,
+template<typename MetricType,
typename StatisticType,
- typename MatType>
-class TreeTraits<RectangleTree<SplitType, DescentType, StatisticType, MatType> >
+ typename MatType,
+ typename SplitType,
+ typename DescentType>
+class TreeTraits<RectangleTree<MetricType, StatisticType, MatType, SplitType,
+ DescentType>>
{
public:
/**
@@ -53,7 +55,7 @@ class TreeTraits<RectangleTree<SplitType, DescentType, StatisticType, MatType> >
static const bool BinaryTree = false;
};
-}; // namespace tree
-}; // namespace mlpack
+} // namespace tree
+} // namespace mlpack
#endif
More information about the mlpack-git
mailing list