BayesianNetwork.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_FACTORED_UTILS_BAYESIAN_NETWORK_HEADER_FILE
#define AI_TOOLBOX_FACTORED_UTILS_BAYESIAN_NETWORK_HEADER_FILE
 
#include <AIToolbox/Factored/Types.hpp>
#include <AIToolbox/Factored/Utils/Core.hpp>
#include <AIToolbox/Factored/Utils/FactoredMatrix.hpp>
 
namespace AIToolbox::Factored {
    class DynamicDecisionNetworkGraph {
        public:
            struct ParentSet {
                PartialKeys agents;
                std::vector<PartialKeys> features;
            };
 
            DynamicDecisionNetworkGraph(State S, Action A);
 
            void push(ParentSet parents);
 
            size_t getId(size_t feature, const State & s, const Action & a) const;
 
            size_t getId(size_t feature, const PartialState & s, const PartialAction & a) const;
 
            size_t getId(size_t feature, size_t parentId, size_t actionId) const;
 
            std::pair<size_t, size_t> getIds(size_t feature, const State & s, const Action & a) const;
 
            std::pair<size_t, size_t> getIds(size_t feature, const PartialState & s, const PartialAction & a) const;
 
            std::pair<size_t, size_t> getIds(size_t feature, size_t j) const;
 
            size_t getSize(size_t feature) const;
 
            size_t getPartialSize(size_t feature) const;
 
            size_t getPartialSize(size_t feature, size_t actionId) const;
 
            const State & getS() const;
 
            const Action & getA() const;
 
            const std::vector<ParentSet> & getParentSets() const;
 
        private:
            State S;
            Action A;
            std::vector<ParentSet> parents_;
            std::vector<std::vector<size_t>> startIds_;
    };
 
    using DDNGraph = DynamicDecisionNetworkGraph;
 
    struct DynamicDecisionNetwork {
        using TransitionMatrix = std::vector<Matrix2D>;
 
        double getTransitionProbability(const State & s, const Action & a, const State & s1) const;
 
        double getTransitionProbability(const PartialState & s, const PartialAction & a, const PartialState & s1) const;
 
        const DDNGraph & graph;
        TransitionMatrix transitions;
    };
 
    using DDN = DynamicDecisionNetwork;
 
    BasisMatrix backProject(const DDN & ddn, const BasisFunction & bf);
    FactoredMatrix2D backProject(const DDN & ddn, const FactoredVector & fv);
}
 
#endif