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

Begin with parameter estimation

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-06-30 21:24:12 +02:00
parent 0a31aa2ff1
commit 71d730d228
8 changed files with 236 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
Network.h
View File

@@ -3,12 +3,16 @@
#include "Node.h"
#include <map>
#include <vector>
namespace bayesnet {
class Network {
private:
map<string, Node*> nodes;
map<string, torch::Tensor> cpds; // Map from CPD key to CPD tensor
map<string, vector<int>> dataset;
Node* root;
vector<string> features;
string className;
int laplaceSmoothing;
bool isCyclic(const std::string&, std::unordered_set<std::string>&, std::unordered_set<std::string>&);
public:
@@ -19,9 +23,8 @@ namespace bayesnet {
void addEdge(const string, const string);
map<string, Node*>& getNodes();
void fit(const vector<vector<int>>&, const vector<int>&, const vector<string>&, const string&);
void buildNetwork(const vector<vector<int>>&, const vector<int>&, const vector<string>&, const string&);
torch::Tensor& getCPD(const string&);
void setCPD(const string&, const torch::Tensor&);
void estimateParameters();
void buildNetwork();
void setRoot(string);
Node* getRoot();
};