MiningProblem.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_FACTORED_BANDIT_MINING_PROBLEM_HEADER_FILE
#define AI_TOOLBOX_FACTORED_BANDIT_MINING_PROBLEM_HEADER_FILE
 
#include <AIToolbox/Factored/Bandit/Types.hpp>
 
namespace AIToolbox::Factored::Bandit {
    class MiningBandit {
        public:
            MiningBandit(Action A, std::vector<unsigned> workersPerVillage, std::vector<double> productivityPerMine, bool normalizeToOne = true);
 
            const Rewards & sampleR(const Action & a) const;
 
            double getRegret(const Action & a) const;
 
            const Action & getOptimalAction() const;
 
            const Action & getA() const;
 
            const std::vector<PartialKeys> & getGroups() const;
 
            std::vector<QFunctionRule> getDeterministicRules() const;
 
            double getNormalizationConstant() const;
 
        private:
            void computeProbabilities(const Action & a) const;
 
            Action A;
            std::vector<unsigned> workersPerVillage_;
            std::vector<double> productivityPerMine_;
 
            Action optimal_;
            double rewardNorm_;
            bool normalizeToOne_;
 
            std::vector<PartialKeys> villagesPerMine_;
 
            mutable Rewards helper_;
            mutable AIToolbox::RandomEngine rand_;
    };
 
    std::tuple<Action, std::vector<unsigned>, std::vector<double>> makeMiningParameters(unsigned seed);
}
 
#endif