Refactor library structure
This commit is contained in:
15
bayesnet/classifiers/TAN.h
Normal file
15
bayesnet/classifiers/TAN.h
Normal file
@@ -0,0 +1,15 @@
|
||||
#ifndef TAN_H
|
||||
#define TAN_H
|
||||
#include "Classifier.h"
|
||||
namespace bayesnet {
|
||||
class TAN : public Classifier {
|
||||
private:
|
||||
protected:
|
||||
void buildModel(const torch::Tensor& weights) override;
|
||||
public:
|
||||
TAN();
|
||||
virtual ~TAN() = default;
|
||||
std::vector<std::string> graph(const std::string& name = "TAN") const override;
|
||||
};
|
||||
}
|
||||
#endif
|
Reference in New Issue
Block a user