Fix ld that was discretizing all input features

2025-08-19 12:29:54 +02:00
parent 89142f8997
commit 9f9369269a
5 changed files with 32 additions and 23 deletions
--- a/bayesnet/ensembles/AODELd.cc
+++ b/bayesnet/ensembles/AODELd.cc
@@ -19,7 +19,7 @@ namespace bayesnet {
        Xf = X_;
        y = y_;
        // Fills std::vectors Xv & yv with the data from tensors X_ (discretized) & y
-        states = fit_local_discretization(y);
+        states = fit_local_discretization(y, states_);
        // We have discretized the input data
        // 1st we need to fit the model to build the normal AODE structure, Ensemble::fit  
        // calls buildModel to initialize the base models