[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