Set smoothing as fit parameter

2024-06-11 11:40:45 +02:00
parent 27a3e5a5e0
commit b34869cc61
30 changed files with 168 additions and 178 deletions
--- a/bayesnet/ensembles/Ensemble.h
+++ b/bayesnet/ensembles/Ensemble.h
@@ -46,7 +46,7 @@ namespace bayesnet {
        unsigned n_models;
        std::vector<std::unique_ptr<Classifier>> models;
        std::vector<double> significanceModels;
-        void trainModel(const torch::Tensor& weights) override;
+        void trainModel(const torch::Tensor& weights, const Smoothing_t smoothing) override;
        bool predict_voting;
    };
 }