diff --git a/bayesnet/classifiers/TANLd.cc b/bayesnet/classifiers/TANLd.cc index 32bd7b8..b415b0f 100644 --- a/bayesnet/classifiers/TANLd.cc +++ b/bayesnet/classifiers/TANLd.cc @@ -8,7 +8,10 @@ #include 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& features_, const std::string& className_, map>& states_, const Smoothing_t smoothing) { checkInput(X_, y_);