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

Implement predict and predict_proba

Browse Source 
Add samples and add parameters to main
This commit is contained in:
Ricardo Montañana Gómez
2023-07-01 14:45:44 +02:00
parent 79e7912ab3
commit 23f0b0f55c
13 changed files with 34538 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/Network.h
View File

@@ -15,6 +15,7 @@ namespace bayesnet {
string className;
int laplaceSmoothing;
bool isCyclic(const std::string&, std::unordered_set<std::string>&, std::unordered_set<std::string>&);
pair<int, double> predict_sample(const vector<int>&);
public:
Network();
Network(int);
@@ -24,9 +25,11 @@ namespace bayesnet {
map<string, Node*>& getNodes();
void fit(const vector<vector<int>>&, const vector<int>&, const vector<string>&, const string&);
void estimateParameters();
void buildNetwork();
void setRoot(string);
Node* getRoot();
vector<int> predict(const vector<vector<int>>&);
vector<pair<int, double>> predict_proba(const vector<vector<int>>&);
double score(const vector<vector<int>>&, const vector<int>&);
};
}
#endif