refactor to use in python

2023-07-12 01:05:24 +02:00
parent d1eaab6408
commit a60b06e2f2
4 changed files with 33 additions and 5 deletions
--- a/src/Metrics.hpp
+++ b/src/Metrics.hpp
@@ -7,7 +7,7 @@ using namespace std;
 namespace bayesnet {
    class Metrics {
    private:
-        torch::Tensor& samples;
+        torch::Tensor samples;
        vector<string>& features;
        string& className;
        int classNumStates;
@@ -17,7 +17,8 @@ namespace bayesnet {
        double mutualInformation(torch::Tensor&, torch::Tensor&);
    public:
        Metrics(torch::Tensor&, vector<string>&, string&, int);
-        torch::Tensor conditionalEdgeWeight();
+        Metrics(vector<vector<int>>&, int, int, vector<string>&, string&, int);
+        vector<float> conditionalEdgeWeights();
    };
 }
 #endif