Add significanceModels to AODELd
This commit is contained in:
parent
1c6af619b5
commit
284ef6dfd1
11
.vscode/launch.json
vendored
11
.vscode/launch.json
vendored
@ -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",
|
||||
},
|
||||
|
@ -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)
|
||||
{
|
||||
|
Loading…
Reference in New Issue
Block a user