Adam.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_ADAM_HEADER_FILE
#define AI_TOOLBOX_ADAM_HEADER_FILE
 
#include <AIToolbox/Types.hpp>
 
namespace AIToolbox {
    class Adam {
        public:
            Adam(AIToolbox::Vector * point, const AIToolbox::Vector & gradient, double alpha = 0.001, double beta1 = 0.9, double beta2 = 0.999, double epsilon = 1e-8);
 
            void step();
 
            void reset();
 
            void reset(AIToolbox::Vector * point, const AIToolbox::Vector & gradient);
 
            void setAlpha(double alpha);
 
            void setBeta1(double beta1);
 
            void setBeta2(double beta2);
 
            void setEpsilon(double epsilon);
 
            double getAlpha() const;
 
            double getBeta1() const;
 
            double getBeta2() const;
 
            double getEpsilon() const;
 
        private:
            AIToolbox::Vector * point_;
            const AIToolbox::Vector * gradient_;
            AIToolbox::Vector m_, v_;
 
            double beta1_, beta2_, alpha_, epsilon_;
            unsigned step_;
    };
}
 
#endif