[mlpack-git] master: Always normalise a projection vector. (e81731d)

gitdub at mlpack.org gitdub at mlpack.org
Thu Aug 11 08:57:40 EDT 2016


Repository : https://github.com/mlpack/mlpack
On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/0f4b25acd6aaa14294c044874ba6cc0751712baa...0a19d07bd39e6223991976474bc79671ba8aa0f0

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

commit e81731d94d36c4231d045114443ef4f95946308c
Author: MarcosPividori <marcos.pividori at gmail.com>
Date:   Thu Aug 11 09:57:23 2016 -0300

    Always normalise a projection vector.


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

e81731d94d36c4231d045114443ef4f95946308c
 src/mlpack/core/tree/space_split/projection_vector.hpp | 4 ++--
 src/mlpack/core/tree/space_split/space_split_impl.hpp  | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/mlpack/core/tree/space_split/projection_vector.hpp b/src/mlpack/core/tree/space_split/projection_vector.hpp
index 4b0b2b9..5ade0f9 100644
--- a/src/mlpack/core/tree/space_split/projection_vector.hpp
+++ b/src/mlpack/core/tree/space_split/projection_vector.hpp
@@ -8,7 +8,7 @@
 #define MLPACK_CORE_TREE_SPILL_TREE_PROJECTION_VECTOR_HPP
 
 #include <mlpack/core.hpp>
-
+#include "../bounds.hpp"
 namespace mlpack {
 namespace tree {
 
@@ -102,7 +102,7 @@ class ProjVector
    * @param vect Vector to be considered.
    */
   ProjVector(const arma::vec& vect) :
-      projVect(vect)
+      projVect(arma::normalise(vect))
   {};
 
   /**
diff --git a/src/mlpack/core/tree/space_split/space_split_impl.hpp b/src/mlpack/core/tree/space_split/space_split_impl.hpp
index 96b4235..94f82d8 100644
--- a/src/mlpack/core/tree/space_split/space_split_impl.hpp
+++ b/src/mlpack/core/tree/space_split/space_split_impl.hpp
@@ -88,7 +88,7 @@ bool SpaceSplit<MetricType, MatType>::GetProjVector(
     return false;
 
   // Calculate the normalized projection vector.
-  projVector = ProjVector(arma::normalise(data.col(snd) - data.col(fst)));
+  projVector = ProjVector(data.col(snd) - data.col(fst));
 
   arma::vec midPoint = (data.col(snd) + data.col(fst)) / 2;
 




More information about the mlpack-git mailing list