rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests 1 Actions Packages Projects Releases 6 Wiki Activity

Add cycle detect adding edges

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-06-29 23:53:33 +02:00
parent d59bf03a51
commit 31c22898de
5 changed files with 87 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
Node.h
View File

@@ -15,14 +15,16 @@ namespace bayesnet {
int numStates;
torch::Tensor cpt;
public:
Node(const std::string& name, int numStates);
void addParent(Node* parent);
void addChild(Node* child);
Node(const std::string&, int);
void addParent(Node*);
void addChild(Node*);
void removeParent(Node*);
void removeChild(Node*);
string getName() const;
vector<Node*>& getParents();
vector<Node*>& getChildren();
torch::Tensor& getCPT();
void setCPT(const torch::Tensor& cpt);
void setCPT(const torch::Tensor&);
int getNumStates() const;
int getId() const { return id; }
string getCPDKey(const Node*) const;