[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