Fix TANLd valid_hyperparameters

2025-07-08 17:28:27 +02:00
parent e5227c5f4b
commit aa77745e55
1 changed files with 4 additions and 1 deletions
--- a/bayesnet/classifiers/TANLd.cc
+++ b/bayesnet/classifiers/TANLd.cc
@@ -8,7 +8,10 @@
 #include <memory>

 namespace bayesnet {
-    TANLd::TANLd() : TAN(), Proposal(dataset, features, className) {}
+    TANLd::TANLd() : TAN(), Proposal(dataset, features, className)
+    {
+        validHyperparameters = validHyperparameters_ld; // Inherits the valid hyperparameters from Proposal
+    }
    TANLd& TANLd::fit(torch::Tensor& X_, torch::Tensor& y_, const std::vector<std::string>& features_, const std::string& className_, map<std::string, std::vector<int>>& states_, const Smoothing_t smoothing)
    {
        checkInput(X_, y_);