MaxPlus.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_FACTORED_BANDIT_MAX_PLUS_HEADER_FILE
#define AI_TOOLBOX_FACTORED_BANDIT_MAX_PLUS_HEADER_FILE
 
#include <AIToolbox/Factored/Bandit/Types.hpp>
#include <AIToolbox/Factored/Utils/FactorGraph.hpp>
#include <AIToolbox/Factored/Utils/Core.hpp>
 
namespace AIToolbox::Factored::Bandit {
    class MaxPlus {
        public:
            using Result = std::tuple<Action, double>;
 
            // Values of factor (in theory N dimensional matrix)
            using Graph = FactorGraph<Vector>;
 
            MaxPlus(unsigned iterations = 10);
 
            Result operator()(const Action & A, const Graph & graph);
 
            unsigned getIterations() const;
 
            void setIterations(unsigned iterations);
 
        private:
            unsigned iterations_;
    };
}
 
#endif