constant_step.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_OPTIMIZERS_PARALLEL_SGD_CONSTANT_STEP_HPP
#define MLPACK_CORE_OPTIMIZERS_PARALLEL_SGD_CONSTANT_STEP_HPP

#include <mlpack/prereqs.hpp>

namespace mlpack {
namespace optimization {

class ConstantStep
{
 public:
  ConstantStep(const double step = 0.01) : step(step) { /* Nothing to do */ }

  double StepSize(const size_t /* numEpoch */)
  {
    return step;
  }
 private:
  double step;
};

} // namespace optimization
} // namespace mlpack

#endif