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