52 lines
1.1 KiB
C++
52 lines
1.1 KiB
C++
#include "Node.h"
|
|
|
|
namespace bayesnet {
|
|
int Node::next_id = 0;
|
|
|
|
Node::Node(const std::string& name, int numStates)
|
|
: id(next_id++), name(name), numStates(numStates), cpt(torch::Tensor()), parents(vector<Node*>()), children(vector<Node*>())
|
|
{
|
|
}
|
|
|
|
string Node::getName() const
|
|
{
|
|
return name;
|
|
}
|
|
|
|
void Node::addParent(Node* parent)
|
|
{
|
|
parents.push_back(parent);
|
|
}
|
|
void Node::removeParent(Node* parent)
|
|
{
|
|
parents.erase(std::remove(parents.begin(), parents.end(), parent), parents.end());
|
|
}
|
|
void Node::removeChild(Node* child)
|
|
{
|
|
children.erase(std::remove(children.begin(), children.end(), child), children.end());
|
|
}
|
|
void Node::addChild(Node* child)
|
|
{
|
|
children.push_back(child);
|
|
}
|
|
vector<Node*>& Node::getParents()
|
|
{
|
|
return parents;
|
|
}
|
|
vector<Node*>& Node::getChildren()
|
|
{
|
|
return children;
|
|
}
|
|
int Node::getNumStates() const
|
|
{
|
|
return numStates;
|
|
}
|
|
torch::Tensor& Node::getCPT()
|
|
{
|
|
return cpt;
|
|
}
|
|
void Node::setCPT(const torch::Tensor& cpt)
|
|
{
|
|
this->cpt = cpt;
|
|
}
|
|
} |