LRPPolicy.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_BANDIT_LRP_POLICY_HEADER_FILE
#define AI_TOOLBOX_BANDIT_LRP_POLICY_HEADER_FILE
 
#include <AIToolbox/Types.hpp>
#include <AIToolbox/Bandit/Policies/PolicyInterface.hpp>
 
namespace AIToolbox::Bandit {
    class LRPPolicy : public PolicyInterface {
        public:
            LRPPolicy(size_t A, double a, double b = 0.0);
 
            void stepUpdateP(size_t a, bool result);
 
            virtual size_t sampleAction() const override;
 
            virtual double getActionProbability(const size_t & a) const override;
 
            void setAParam(double a);
 
            double getAParam() const;
 
            void setBParam(double b);
 
            double getBParam() const;
 
            virtual Vector getPolicy() const override;
 
        private:
            double a_, invB_, divB_;
            Vector policy_;
    };
}
 
#endif