FlattenedModel.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_FACTORED_BANDIT_FLATTENED_MODEL_HEADER_FILE
#define AI_TOOLBOX_FACTORED_BANDIT_FLATTENED_MODEL_HEADER_FILE
 
#include <AIToolbox/Factored/Bandit/Model.hpp>
 
namespace AIToolbox::Factored::Bandit {
    template <typename Dist>
    class FlattenedModel {
        public:
            FlattenedModel(const Model<Dist> & model);
 
            double sampleR(size_t a) const;
 
            Action convertA(size_t a) const;
 
            size_t getA() const;
 
            const Model<Dist> & getModel() const;
 
        private:
            const Model<Dist> & model_;
 
            size_t A;
            mutable Action helper_;
    };
 
    template <typename Dist>
    FlattenedModel<Dist>::FlattenedModel(const Model<Dist> & model) :
            model_(model), A(factorSpace(model.getA())), helper_(model.getA().size())
    {}
 
    template <typename Dist>
    double FlattenedModel<Dist>::sampleR(size_t a) const {
        toFactors(model_.getA(), a, &helper_);
        return model_.sampleR(helper_).sum();
    }
 
    template <typename Dist>
    size_t FlattenedModel<Dist>::getA() const { return A; }
    template <typename Dist>
    const Model<Dist> & FlattenedModel<Dist>::getModel() const { return model_; }
}
 
#endif