Add getStates method

src/Network.cc

@@ -38,6 +38,14 @@ namespace bayesnet {
     {
         return classNumStates;
     }
+    int Network::getStates()
+    {
+        int result = 0;
+        for (auto node : nodes) {
+            result += node.second->getNumStates();
+        }
+        return result;
+    }
     string Network::getClassName()
     {
         return className;

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&);