32 #ifndef ADAPTMOMEMTUMCORRECTION_H
33 #define ADAPTMOMEMTUMCORRECTION_H
66 virtual const char*
get_name()
const {
return "AdaptMomentumCorrection"; }
virtual DescendPair get_corrected_descend_direction(float64_t negative_descend_direction, index_t idx)
SGVector< float64_t > m_descend_rate
virtual void set_adapt_rate(float64_t adapt_rate, float64_t rate_min=0.0, float64_t rate_max=CMath::INFTY)
float64_t m_init_descend_rate
static const float64_t INFTY
infinity
virtual void set_momentum_correction(MomentumCorrection *correction)
virtual void set_init_descend_rate(float64_t init_descend_rate)
virtual ~AdaptMomentumCorrection()
virtual const char * get_name() const
virtual bool is_initialized()
MomentumCorrection * m_momentum_correction
AdaptMomentumCorrection()
virtual void initialize_previous_direction(index_t len)
all of classes and functions are contained in the shogun namespace
This is a base class for momentum correction methods.
This implements the adaptive momentum correction method.
virtual void set_correction_weight(float64_t weight)