Refactor constructor

2023-07-12 03:07:10 +02:00
parent a60b06e2f2
commit 5793c31bc4
3 changed files with 16 additions and 14 deletions
--- a/src/Metrics.hpp
+++ b/src/Metrics.hpp
@@ -17,7 +17,7 @@ namespace bayesnet {
        double mutualInformation(torch::Tensor&, torch::Tensor&);
    public:
        Metrics(torch::Tensor&, vector<string>&, string&, int);
-        Metrics(vector<vector<int>>&, int, int, vector<string>&, string&, int);
+        Metrics(vector<vector<int>>&, vector<int>&, vector<string>&, string&, int);
        vector<float> conditionalEdgeWeights();
    };
 }