Experience.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_BANDIT_EXPERIENCE_HEADER_FILE
#define AI_TOOLBOX_BANDIT_EXPERIENCE_HEADER_FILE
 
#include <AIToolbox/Bandit/Types.hpp>
 
namespace AIToolbox::Bandit {
    class Experience {
        public:
            using VisitsTable = std::vector<unsigned long>;
 
            Experience(size_t A);
 
            void record(size_t a, double rew);
 
            void reset();
 
            unsigned long getTimesteps() const;
 
            const QFunction & getRewardMatrix() const;
 
            const VisitsTable & getVisitsTable() const;
 
            const Vector & getM2Matrix() const;
 
            size_t getA() const;
 
        private:
            QFunction q_;
            Vector M2s_;
            VisitsTable counts_;
            unsigned long timesteps_;
    };
}
 
#endif