ReusingIterativeLocalSearch.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_FACTORED_BANDIT_REUSING_ITERATIVE_LOCAL_SEARCH_HEADER_FILE
#define AI_TOOLBOX_FACTORED_BANDIT_REUSING_ITERATIVE_LOCAL_SEARCH_HEADER_FILE
 
#include <AIToolbox/Factored/Bandit/Types.hpp>
#include <AIToolbox/Factored/Utils/Core.hpp>
#include <AIToolbox/Factored/Utils/FactorGraph.hpp>
#include <AIToolbox/Factored/Bandit/Algorithms/Utils/LocalSearch.hpp>
 
namespace AIToolbox::Factored::Bandit {
    class ReusingIterativeLocalSearch {
        public:
            using Result = std::tuple<Action, double>;
            using Graph = LocalSearch::Graph;
 
            ReusingIterativeLocalSearch(double resetActionProbability = 0.3, double randomizeFactorProbability = 0.1, unsigned trialNum = 10, bool forceResetAction = true);
 
            Result operator()(const Action & A, const Graph & graph);
 
            double getResetActionProbability() const;
 
            void setResetActionProbability(double resetActionProbability);
 
            double getRandomizeFactorProbability() const;
 
            void setRandomizeFactorProbability(double randomizeFactorProbability);
 
            unsigned getTrialNum() const;
 
            void setTrialNum(unsigned trialNum);
 
            bool getForceResetAction() const;
 
            void setForceResetAction(bool forceResetAction);
 
        private:
            // Parameters
            double resetActionProbability_;
            double randomizeFactorProbability_;
            unsigned trialNum_;
            bool forceResetAction_;
 
            // Caches
            Action action_, newAction_;
 
            // Nested local search
            LocalSearch ls_;
 
            RandomEngine rnd_;
    };
};
 
#endif