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

Implement Cestnik & Laplace smoothing

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-06-09 17:19:38 +02:00
parent 6d9badc33b
commit 684443a788
9 changed files with 28 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
bayesnet/network/Node.h
View File

@@ -23,7 +23,7 @@ namespace bayesnet {
std::vector<Node*>& getParents();
std::vector<Node*>& getChildren();
torch::Tensor& getCPT();
void computeCPT(const torch::Tensor& dataset, const std::vector<std::string>& features, const double laplaceSmoothing, const torch::Tensor& weights);
void computeCPT(const torch::Tensor& dataset, const std::vector<std::string>& features, const double smoothing, const torch::Tensor& weights);
int getNumStates() const;
void setNumStates(int);
unsigned minFill();