Begin with parameter estimation

2023-06-30 21:24:12 +02:00
parent 0a31aa2ff1
commit 71d730d228
8 changed files with 236 additions and 87 deletions
--- a/Node.h
+++ b/Node.h
@@ -12,11 +12,12 @@ namespace bayesnet {
        string name;
        vector<Node*> parents;
        vector<Node*> children;
+        torch::Tensor cpTable;
        int numStates;
        torch::Tensor cpt;
    public:
        Node(const std::string&, int);
-        void addParent(Node*);        
+        void addParent(Node*);
        void addChild(Node*);
        void removeParent(Node*);
        void removeChild(Node*);
@@ -27,7 +28,6 @@ namespace bayesnet {
        void setCPT(const torch::Tensor&);
        int getNumStates() const;
        int getId() const { return id; }
-        string getCPDKey(const Node*) const;
    };
 }
 #endif