[mlpack-git] master: Always resize the labels vector when Classify() is called! (cb948db)

gitdub at big.cc.gt.atl.ga.us gitdub at big.cc.gt.atl.ga.us
Mon Nov 30 17:24:13 EST 2015


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

On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/10b9d45b806a3e879b0564d78ccb183ebc7051ba...31c557d9cc7e4da57fd8a246085c19e076d12271

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

commit cb948db156cd0bc6150abdd2a5457da56e8794a8
Author: Ryan Curtin <ryan at ratml.org>
Date:   Sat Nov 21 01:40:49 2015 +0000

    Always resize the labels vector when Classify() is called!


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

cb948db156cd0bc6150abdd2a5457da56e8794a8
 src/mlpack/methods/decision_stump/decision_stump_impl.hpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/mlpack/methods/decision_stump/decision_stump_impl.hpp b/src/mlpack/methods/decision_stump/decision_stump_impl.hpp
index 446ce2e..c4b8b77 100644
--- a/src/mlpack/methods/decision_stump/decision_stump_impl.hpp
+++ b/src/mlpack/methods/decision_stump/decision_stump_impl.hpp
@@ -128,6 +128,7 @@ template<typename MatType>
 void DecisionStump<MatType>::Classify(const MatType& test,
                                       arma::Row<size_t>& predictedLabels)
 {
+  predictedLabels.set_size(test.n_cols);
   for (size_t i = 0; i < test.n_cols; i++)
   {
     // Determine which bin the test point falls into.



More information about the mlpack-git mailing list