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 AODE_H
       8              : #define AODE_H
       9              : #include "bayesnet/classifiers/SPODE.h"
      10              : #include "Ensemble.h"
      11              : namespace bayesnet {
      12              :     class AODE : public Ensemble {
      13              :     public:
      14              :         AODE(bool predict_voting = false);
      15           14 :         virtual ~AODE() {};
      16              :         void setHyperparameters(const nlohmann::json& hyperparameters) override;
      17              :         std::vector<std::string> graph(const std::string& title = "AODE") const override;
      18              :     protected:
      19              :         void buildModel(const torch::Tensor& weights) override;
      20              :     };
      21              : }
      22              : #endif