diff --git a/src/mlpack/methods/sparse_coding/sparse_coding_impl.hpp b/src/mlpack/methods/sparse_coding/sparse_coding_impl.hpp index 7919c20198..08ac5b79ca 100644 --- a/src/mlpack/methods/sparse_coding/sparse_coding_impl.hpp +++ b/src/mlpack/methods/sparse_coding/sparse_coding_impl.hpp @@ -255,7 +255,7 @@ double SparseCoding::OptimizeDictionary( << "." << std::endl; Log::Debug << " Improvement: " << std::scientific << improvement << ".\n"; - if (improvement < newtonTolerance) + if (normGradient < newtonTolerance) converged = true; } @@ -292,7 +292,7 @@ double SparseCoding::OptimizeDictionary( } } } - //printf("final reconstruction error: %e\n", norm(data - dictionary * codes, "fro")); + return normGradient; }