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

Add tests to 90% coverage

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2025-03-14 14:53:22 +01:00
parent c234308701
commit 400967b4e3
14 changed files with 943 additions and 463 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
bayesnet/ensembles/Boost.h
View File

@@ -34,6 +34,11 @@ namespace bayesnet {
void buildModel(const torch::Tensor& weights) override;
std::tuple<torch::Tensor&, double, bool> update_weights(torch::Tensor& ytrain, torch::Tensor& ypred, torch::Tensor& weights);
std::tuple<torch::Tensor&, double, bool> update_weights_block(int k, torch::Tensor& ytrain, torch::Tensor& weights);
void add_model(std::unique_ptr<Classifier> model, double significance);
void remove_last_model();
//
// Attributes
//
torch::Tensor X_train, y_train, X_test, y_test;
// Hyperparameters
bool bisection = true; // if true, use bisection stratety to add k models at once to the ensemble
@@ -49,4 +54,4 @@ namespace bayesnet {
bool alpha_block = false; // if true, the alpha is computed with the ensemble built so far and the new model
};
}
#endif
#endif