PolicyWrapper.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_MDP_POLICY_WRAPPER_HEADER_FILE
#define AI_TOOLBOX_MDP_POLICY_WRAPPER_HEADER_FILE
 
#include <vector>
#include <tuple>
 
#include <AIToolbox/MDP/Types.hpp>
#include <AIToolbox/MDP/Policies/PolicyInterface.hpp>
 
namespace AIToolbox::MDP {
    class PolicyWrapper : public PolicyInterface {
        public:
            using PolicyMatrix = Matrix2D;
 
            PolicyWrapper(const PolicyMatrix & p);
 
            virtual size_t sampleAction(const size_t & s) const override;
 
            virtual double getActionProbability(const size_t & s, const size_t & a) const override;
 
            const PolicyMatrix & getPolicyMatrix() const;
 
            virtual Matrix2D getPolicy() const override;
 
        private:
            const PolicyMatrix & policy_;
    };
}
 
#endif