Add selectKParis method

2024-05-16 11:17:21 +02:00
parent 8784a24898
commit 2e3e0e0fc2
8 changed files with 224 additions and 26 deletions
--- a/bayesnet/ensembles/BoostA2DE.h
+++ b/bayesnet/ensembles/BoostA2DE.h
@@ -18,6 +18,8 @@ namespace bayesnet {
        std::vector<std::string> graph(const std::string& title = "BoostA2DE") const override;
    protected:
        void trainModel(const torch::Tensor& weights) override;
+    private:
+        std::vector<int> initializeModels();
    };
 }
 #endif