Fix classifier build in proposal

2025-07-07 02:10:08 +02:00
parent 0ce7f664b4
commit 2c7352ac38
7 changed files with 29 additions and 16 deletions
--- a/bayesnet/classifiers/Proposal.h
+++ b/bayesnet/classifiers/Proposal.h
@@ -31,7 +31,7 @@ namespace bayesnet {
        map<std::string, std::vector<int>> iterativeLocalDiscretization(
            const torch::Tensor& y,
            Classifier* classifier,
-            const torch::Tensor& dataset,
+            torch::Tensor& dataset,
            const std::vector<std::string>& features,
            const std::string& className,
            const map<std::string, std::vector<int>>& initialStates,