rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests 1 Actions Packages Projects Releases 6 Wiki Activity

Create hyperparameter block_update

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-04-08 23:36:05 +02:00
parent f9553a38d7
commit da2a969686
4 changed files with 38 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
bayesnet/ensembles/BoostAODE.cc
View File

@@ -16,7 +16,7 @@ namespace bayesnet {
{
validHyperparameters = {
"maxModels", "bisection", "order", "convergence", "threshold",
"select_features", "maxTolerance", "predict_voting"
"select_features", "maxTolerance", "predict_voting", "block_update"
};
}
@@ -94,6 +94,10 @@ namespace bayesnet {
}
hyperparameters.erase("select_features");
}
if (hyperparameters.contains("block_update")) {
block_update = hyperparameters["block_update"];
hyperparameters.erase("block_update");
}
Classifier::setHyperparameters(hyperparameters);
}
std::tuple<torch::Tensor&, double, bool> update_weights(torch::Tensor& ytrain, torch::Tensor& ypred, torch::Tensor& weights)