Solved Ld poor results

2023-08-12 11:49:18 +02:00
parent 3a85481a5a
commit 405887f833
7 changed files with 13 additions and 9 deletions
--- a/src/BayesNet/Proposal.h
+++ b/src/BayesNet/Proposal.h
@@ -15,7 +15,7 @@ namespace bayesnet {
    protected:
        torch::Tensor prepareX(torch::Tensor& X);
        void localDiscretizationProposal(map<string, vector<int>>& states, Network& model);
-        void fit_local_discretization(map<string, vector<int>>& states, torch::Tensor& y);
+        map<string, vector<int>> fit_local_discretization(torch::Tensor& y);
        torch::Tensor Xf; // X continuous nxm tensor
        torch::Tensor y; // y discrete nx1 tensor
        map<string, mdlp::CPPFImdlp*> discretizers;