Policy.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_MDP_POLICY_HEADER_FILE
#define AI_TOOLBOX_MDP_POLICY_HEADER_FILE
 
#include <vector>
#include <tuple>
 
#include <AIToolbox/MDP/Types.hpp>
#include <AIToolbox/MDP/Policies/PolicyWrapper.hpp>
 
namespace AIToolbox::MDP {
    class Policy : public PolicyWrapper {
        public:
            using PolicyMatrix = Matrix2D;
 
            Policy(size_t s, size_t a);
 
            Policy(const PolicyInterface::Base & p);
 
            Policy(const PolicyInterface & p);
 
            Policy(size_t s, size_t a, const ValueFunction & v);
 
            Policy(const PolicyMatrix & p);
 
        private:
            PolicyMatrix policy_;
 
            friend std::istream& operator>>(std::istream &is, Policy & p);
    };
 
    std::istream& operator>>(std::istream &is, Policy & p);
}
 
#endif