Almost complete KDB

2023-07-13 03:44:33 +02:00
parent 8b0aa5ccfb
commit 786d781e29
4 changed files with 51 additions and 7 deletions
--- a/src/Metrics.hpp
+++ b/src/Metrics.hpp
@@ -19,6 +19,7 @@ namespace bayesnet {
        Metrics(torch::Tensor&, vector<string>&, string&, int);
        Metrics(const vector<vector<int>>&, const vector<int>&, const vector<string>&, const string&, const int);
        vector<float> conditionalEdgeWeights();
+        torch::Tensor conditionalEdge();
    };
 }
 #endif