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