Refactor hyperparameters classifier management

2024-04-08 00:55:30 +02:00
parent 0d6a081d01
commit 9014649a0d
8 changed files with 43 additions and 15 deletions
--- a/bayesnet/ensembles/BoostAODE.cc
+++ b/bayesnet/ensembles/BoostAODE.cc
@@ -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)
    {