Refactor library structure and add sample

This commit is contained in:
2024-02-27 13:06:13 +01:00
parent f10d0daf2e
commit 903b143338
49 changed files with 104 additions and 8 deletions

18
src/classifiers/TANLd.h Normal file
View File

@@ -0,0 +1,18 @@
#ifndef TANLD_H
#define TANLD_H
#include "TAN.h"
#include "Proposal.h"
namespace bayesnet {
class TANLd : public TAN, public Proposal {
private:
public:
TANLd();
virtual ~TANLd() = default;
TANLd& fit(torch::Tensor& X, torch::Tensor& y, const std::vector<std::string>& features, const std::string& className, map<std::string, std::vector<int>>& states) override;
std::vector<std::string> graph(const std::string& name = "TAN") const override;
torch::Tensor predict(torch::Tensor& X) override;
static inline std::string version() { return "0.0.1"; };
};
}
#endif // !TANLD_H