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

Refactor hyperparameters classifier management

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-04-08 00:55:30 +02:00
parent 0d6a081d01
commit 9014649a0d
8 changed files with 43 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bayesnet/ensembles/AODE.cc
View File

@@ -13,9 +13,7 @@ namespace bayesnet {
predict_voting = hyperparameters["predict_voting"];
hyperparameters.erase("predict_voting");
}
if (!hyperparameters.empty()) {
throw std::invalid_argument("Invalid hyperparameters" + hyperparameters.dump());
}
Classifier::setHyperparameters(hyperparameters);
}
void AODE::buildModel(const torch::Tensor& weights)
{

4
bayesnet/ensembles/BoostAODE.cc
View File

@@ -94,9 +94,7 @@ namespace bayesnet {
}
hyperparameters.erase("select_features");
}
if (!hyperparameters.empty()) {
throw std::invalid_argument("Invalid hyperparameters" + hyperparameters.dump());
}
Classifier::setHyperparameters(hyperparameters);
}
std::tuple<torch::Tensor&, double, bool> update_weights(torch::Tensor& ytrain, torch::Tensor& ypred, torch::Tensor& weights)
{

2
bayesnet/ensembles/BoostAODE.h
View File

@@ -20,7 +20,7 @@ namespace bayesnet {
BoostAODE(bool predict_voting = false);
virtual ~BoostAODE() = default;
std::vector<std::string> graph(const std::string& title = "BoostAODE") const override;
void setHyperparameters(const nlohmann::json& hyperparameters) override;
void setHyperparameters(const nlohmann::json& hyperparameters_) override;
protected:
void buildModel(const torch::Tensor& weights) override;
void trainModel(const torch::Tensor& weights) override;