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/AODE.cc
+++ b/bayesnet/ensembles/AODE.cc
@@ -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)
    {