Implement 3 types of smoothing

2024-06-10 15:49:01 +02:00
parent 684443a788
commit 27a3e5a5e0
11 changed files with 37 additions and 9 deletions
--- a/bayesnet/classifiers/Classifier.cc
+++ b/bayesnet/classifiers/Classifier.cc
@@ -22,6 +22,7 @@ namespace bayesnet {
        auto n_classes = states.at(className).size();
        metrics = Metrics(dataset, features, className, n_classes);
        model.initialize();
+        model.setSmoothing(smoothing);
        buildModel(weights);
        trainModel(weights);
        fitted = true;