Begin with parameter estimation

This commit is contained in:
2023-06-30 21:24:12 +02:00
parent 0a31aa2ff1
commit 71d730d228
8 changed files with 236 additions and 87 deletions

View File

@@ -41,8 +41,12 @@ namespace bayesnet {
{
return numStates;
}
string Node::getCPDKey(const Node* child) const
torch::Tensor& Node::getCPT()
{
return name + "-" + child->getName();
return cpt;
}
void Node::setCPT(const torch::Tensor& cpt)
{
this->cpt = cpt;
}
}