Add significanceModels to AODELd

2023-08-24 12:58:53 +02:00
parent 1c6af619b5
commit 284ef6dfd1
2 changed files with 6 additions and 6 deletions
--- a/src/BayesNet/AODELd.cc
+++ b/src/BayesNet/AODELd.cc
@@ -26,6 +26,7 @@ namespace bayesnet {
            models.push_back(std::make_unique<SPODELd>(i));
        }
        n_models = models.size();
+        significanceModels = vector<double>(n_models, 1.0);
    }
    void AODELd::trainModel(const torch::Tensor& weights)
    {