Line data    Source code

       1              : // ***************************************************************
       2              : // SPDX-FileCopyrightText: Copyright 2024 Ricardo Montañana Gómez
       3              : // SPDX-FileType: SOURCE
       4              : // SPDX-License-Identifier: MIT
       5              : // ***************************************************************
       6              : 
       7              : #ifndef A2DE_H
       8              : #define A2DE_H
       9              : #include "bayesnet/classifiers/SPnDE.h"
      10              : #include "Ensemble.h"
      11              : namespace bayesnet {
      12              :     class A2DE : public Ensemble {
      13              :     public:
      14              :         A2DE(bool predict_voting = false);
      15           12 :         virtual ~A2DE() {};
      16              :         void setHyperparameters(const nlohmann::json& hyperparameters) override;
      17              :         std::vector<std::string> graph(const std::string& title = "A2DE") const override;
      18              :     protected:
      19              :         void buildModel(const torch::Tensor& weights) override;
      20              :     };
      21              : }
      22              : #endif