[mlpack-git] master: Use Row for labels. (9b4453a)

gitdub at big.cc.gt.atl.ga.us gitdub at big.cc.gt.atl.ga.us
Wed Dec 23 11:45:31 EST 2015


Repository : https://github.com/mlpack/mlpack

On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/de9cc4b05069e1fa4793d9355f2f595af5ff45d2...6070527af14296cd99739de6c62666cc5d2a2125

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

commit 9b4453a707103c1e07517e743e78e5b45f4236c1
Author: Ryan Curtin <ryan at ratml.org>
Date:   Fri Oct 30 17:34:40 2015 +0000

    Use Row for labels.


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

9b4453a707103c1e07517e743e78e5b45f4236c1
 src/mlpack/methods/hoeffding_trees/hoeffding_tree.hpp      | 4 ++--
 src/mlpack/methods/hoeffding_trees/hoeffding_tree_impl.hpp | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/mlpack/methods/hoeffding_trees/hoeffding_tree.hpp b/src/mlpack/methods/hoeffding_trees/hoeffding_tree.hpp
index 28bb4e5..2b780f5 100644
--- a/src/mlpack/methods/hoeffding_trees/hoeffding_tree.hpp
+++ b/src/mlpack/methods/hoeffding_trees/hoeffding_tree.hpp
@@ -76,7 +76,7 @@ class HoeffdingTree
    */
   template<typename MatType>
   HoeffdingTree(const MatType& data,
-                const arma::Col<size_t>& labels,
+                const arma::Row<size_t>& labels,
                 const data::DatasetInfo& datasetInfo,
                 const size_t numClasses,
                 const bool batchTraining = true,
@@ -124,7 +124,7 @@ class HoeffdingTree
    */
   template<typename MatType>
   void Train(const MatType& data,
-             const arma::Col<size_t>& labels,
+             const arma::Row<size_t>& labels,
              const bool batchTraining = true);
 
   /**
diff --git a/src/mlpack/methods/hoeffding_trees/hoeffding_tree_impl.hpp b/src/mlpack/methods/hoeffding_trees/hoeffding_tree_impl.hpp
index afa5789..e7c5300 100644
--- a/src/mlpack/methods/hoeffding_trees/hoeffding_tree_impl.hpp
+++ b/src/mlpack/methods/hoeffding_trees/hoeffding_tree_impl.hpp
@@ -22,7 +22,7 @@ HoeffdingTree<
     NumericSplitType,
     CategoricalSplitType
 >::HoeffdingTree(const MatType& data,
-                 const arma::Col<size_t>& labels,
+                 const arma::Row<size_t>& labels,
                  const data::DatasetInfo& datasetInfo,
                  const size_t numClasses,
                  const bool batchTraining,
@@ -117,7 +117,7 @@ void HoeffdingTree<
     NumericSplitType,
     CategoricalSplitType
 >::Train(const MatType& data,
-         const arma::Col<size_t>& labels,
+         const arma::Row<size_t>& labels,
          const bool batchTraining)
 {
   // Not yet implemented.



More information about the mlpack-git mailing list