Complete TAN with Maximum Spanning Tree

2023-07-15 18:31:50 +02:00
parent e311c27d43
commit e3863387bb
7 changed files with 172 additions and 10 deletions
--- a/src/Metrics.hpp
+++ b/src/Metrics.hpp
@@ -22,7 +22,7 @@ namespace bayesnet {
        vector<float> conditionalEdgeWeights();
        Tensor conditionalEdge();
        vector<pair<string, string>> doCombinations(const vector<string>&);
-        vector<pair<int, int>> maximumSpanningTree(int root, Tensor& weights);
+        vector<pair<int, int>> maximumSpanningTree(vector<string> features, Tensor& weights, int root);
    };
 }
 #endif