Use the new optimizer interface.

This commit is contained in:
marcus
2015-11-12 14:50:18 +01:00
parent a410bf56c6
commit 99de0b655f
@@ -34,7 +34,7 @@ class SteepestDescent
* @param mom The momentum coefficient.
*/
SteepestDescent(DecomposableFunctionType& function,
const double lr = 1,
const double lr = 0.5,
const double mom = 0) :
function(function),
lr(lr),
@@ -66,13 +66,12 @@ class SteepestDescent
{
if (gradient.n_elem != 0)
{
DataType outputGradient;
function.Gradient(outputGradient);
DataType outputGradient = function.Gradient();
gradient += outputGradient;
}
else
{
function.Gradient(gradient);
gradient = function.Gradient();
}
}