vanilla_update.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_OPTIMIZERS_SGD_EMPTY_UPDATE_HPP
#define MLPACK_CORE_OPTIMIZERS_SGD_EMPTY_UPDATE_HPP

#include <mlpack/prereqs.hpp>

namespace mlpack {
namespace optimization {

class VanillaUpdate
{
 public:
  void Initialize(const size_t /* rows */, const size_t /* cols */)
  { /* Do nothing. */ }

  void Update(arma::mat& iterate,
              const double stepSize,
              const arma::mat& gradient)
  {
    // Perform the vanilla SGD update.
    iterate -= stepSize * gradient;
  }
};

} // namespace optimization
} // namespace mlpack

#endif