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 TAN_H
       8              : #define TAN_H
       9              : #include "Classifier.h"
      10              : namespace bayesnet {
      11              :     class TAN : public Classifier {
      12              :     private:
      13              :     protected:
      14              :         void buildModel(const torch::Tensor& weights) override;
      15              :     public:
      16              :         TAN();
      17           38 :         virtual ~TAN() = default;
      18              :         std::vector<std::string> graph(const std::string& name = "TAN") const override;
      19              :     };
      20              : }
      21              : #endif