Add getNumberOfNodes & getNumberOfEdges to Models

Add some more tests
2023-07-19 15:05:44 +02:00
parent 1a21015492
commit 2f5bd0ea7e
8 changed files with 66 additions and 6 deletions
--- a/src/Network.h
+++ b/src/Network.h
@@ -36,6 +36,7 @@ namespace bayesnet {
        map<string, std::unique_ptr<Node>>& getNodes();
        vector<string> getFeatures();
        int getStates();
+        vector<pair<string, string>> getEdges();
        int getClassNumStates();
        string getClassName();
        void fit(const vector<vector<int>>&, const vector<int>&, const vector<string>&, const string&);