diff --git a/.vscode/launch.json b/.vscode/launch.json index 4ac137c..407feb6 100644 --- a/.vscode/launch.json +++ b/.vscode/launch.json @@ -10,7 +10,7 @@ "-d", "iris", "-m", - "KDB", + "TANLd", "-s", "271", "-p", @@ -25,15 +25,14 @@ "program": "${workspaceFolder}/build/src/Platform/main", "args": [ "-m", - "BoostAODE", + "AODELd", "-p", "/Users/rmontanana/Code/discretizbench/datasets", - "--discretize", "--stratified", "-d", - "glass", - "--hyperparameters", - "{\"repeatSparent\": true, \"maxModels\": 12}" + "wine" + // "--hyperparameters", + // "{\"repeatSparent\": true, \"maxModels\": 12}" ], "cwd": "/Users/rmontanana/Code/discretizbench", }, diff --git a/src/BayesNet/AODELd.cc b/src/BayesNet/AODELd.cc index 1a3bd75..6f80715 100644 --- a/src/BayesNet/AODELd.cc +++ b/src/BayesNet/AODELd.cc @@ -26,6 +26,7 @@ namespace bayesnet { models.push_back(std::make_unique(i)); } n_models = models.size(); + significanceModels = vector(n_models, 1.0); } void AODELd::trainModel(const torch::Tensor& weights) {