Add getStates method

2023-07-09 16:25:24 +02:00
parent c4836bd5e3
commit 3750662f2c
2 changed files with 9 additions and 0 deletions
--- a/src/Network.h
+++ b/src/Network.h
@@ -30,6 +30,7 @@ namespace bayesnet {
        void addEdge(const string, const string);
        map<string, Node*>& getNodes();
        vector<string> getFeatures();
+        int getStates();
        int getClassNumStates();
        string getClassName();
        void fit(const vector<vector<int>>&, const vector<int>&, const vector<string>&, const string&);