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/KDBLd.cc
+++ b/bayesnet/classifiers/KDBLd.cc
@@ -37,6 +37,7 @@ namespace bayesnet {
        className = className_;
        states = iterativeLocalDiscretization(y, static_cast<KDB*>(this), dataset, features, className, states_, smoothing);
        KDB::fit(dataset, features, className, states, smoothing);
+        fitted = true;
        return *this;
    }
    torch::Tensor KDBLd::predict(torch::Tensor& X)