Add significanceModels to AODELd

This commit is contained in:
Ricardo Montañana Gómez 2023-08-24 12:58:53 +02:00
parent 1c6af619b5
commit 284ef6dfd1
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
2 changed files with 6 additions and 6 deletions

11
.vscode/launch.json vendored
View File

@ -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",
},

View File

@ -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)
{