Extract buildModel method to parent class in Boost

2024-05-15 20:00:44 +02:00
parent 54496c68f1
commit 8784a24898
6 changed files with 35 additions and 66 deletions
--- a/bayesnet/ensembles/BoostAODE.h
+++ b/bayesnet/ensembles/BoostAODE.h
@@ -17,7 +17,6 @@ namespace bayesnet {
        virtual ~BoostAODE() = default;
        std::vector<std::string> graph(const std::string& title = "BoostAODE") const override;
    protected:
-        void buildModel(const torch::Tensor& weights) override;
        void trainModel(const torch::Tensor& weights) override;
    private:
        std::vector<int> initializeModels();