Complete proposal with only discretizing numeric features

2025-08-24 01:09:25 +02:00
parent 0c7452e35b
commit 7c01646726
9 changed files with 380 additions and 48883 deletions
--- a/bayesnet/classifiers/TANLd.cc
+++ b/bayesnet/classifiers/TANLd.cc
@@ -35,6 +35,7 @@ namespace bayesnet {
        className = className_;
        states = iterativeLocalDiscretization(y, static_cast<TAN*>(this), dataset, features, className, states_, smoothing);
        TAN::fit(dataset, features, className, states, smoothing);
+        fitted = true;
        return *this;
    }
    torch::Tensor TANLd::predict(torch::Tensor& X)