[mlpack-git] master: Minor fix in gradien descent (f8b2c15)

gitdub at mlpack.org gitdub at mlpack.org
Sun Oct 2 10:45:18 EDT 2016


Repository : https://github.com/mlpack/mlpack
On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/998bb2fae41210d03ddf007b51d994a9cf6262cf...5fd539549c181e602e48c18fd9c024382c42d676

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

commit f8b2c154a39be5f3967367ff38b3c8a327a83a03
Author: sumedhghaisas <sumedhghaisas at gmail.com>
Date:   Sun Oct 2 20:15:18 2016 +0530

    Minor fix in gradien descent


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

f8b2c154a39be5f3967367ff38b3c8a327a83a03
 src/mlpack/core/optimizers/gradient_descent/gradient_descent_impl.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/mlpack/core/optimizers/gradient_descent/gradient_descent_impl.hpp b/src/mlpack/core/optimizers/gradient_descent/gradient_descent_impl.hpp
index 564fcdd..22143f3 100644
--- a/src/mlpack/core/optimizers/gradient_descent/gradient_descent_impl.hpp
+++ b/src/mlpack/core/optimizers/gradient_descent/gradient_descent_impl.hpp
@@ -35,7 +35,7 @@ double GradientDescent<FunctionType>::Optimize(
   double lastObjective = DBL_MAX;
 
   // Now iterate!
-  arma::mat gradient(iterate.n_cols);
+  arma::mat gradient(iterate.n_rows, iterate.n_cols);
   for (size_t i = 1; i != maxIterations; ++i)
   {
     // Output current objective function.




More information about the mlpack-git mailing list