rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests 1 Actions Packages Projects Releases 6 Wiki Activity

Add predict_proba to Ld classifiers

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2025-05-12 19:47:04 +02:00
parent 8a02a3a5cb
commit b11620bbe8
12 changed files with 116 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
bayesnet/classifiers/KDBLd.cc
View File

@@ -28,6 +28,11 @@ namespace bayesnet {
auto Xt = prepareX(X);
return KDB::predict(Xt);
}
torch::Tensor KDBLd::predict_proba(torch::Tensor& X)
{
auto Xt = prepareX(X);
return KDB::predict_proba(Xt);
}
std::vector<std::string> KDBLd::graph(const std::string& name) const
{
return KDB::graph(name);

1
bayesnet/classifiers/KDBLd.h
View File

@@ -18,6 +18,7 @@ namespace bayesnet {
KDBLd& fit(torch::Tensor& X, torch::Tensor& y, const std::vector<std::string>& features, const std::string& className, map<std::string, std::vector<int>>& states, const Smoothing_t smoothing) override;
std::vector<std::string> graph(const std::string& name = "KDB") const override;
torch::Tensor predict(torch::Tensor& X) override;
torch::Tensor predict_proba(torch::Tensor& X) override;
static inline std::string version() { return "0.0.1"; };
};
}

2
bayesnet/classifiers/Proposal.cc
View File

@@ -11,7 +11,7 @@ namespace bayesnet {
Proposal::~Proposal()
{
for (auto& [key, value] : discretizers) {
delete value;
delete value;
}
}
void Proposal::checkInput(const torch::Tensor& X, const torch::Tensor& y)

5
bayesnet/classifiers/SPODELd.cc
View File

@@ -43,6 +43,11 @@ namespace bayesnet {
auto Xt = prepareX(X);
return SPODE::predict(Xt);
}
torch::Tensor SPODELd::predict_proba(torch::Tensor& X)
{
auto Xt = prepareX(X);
return SPODE::predict_proba(Xt);
}
std::vector<std::string> SPODELd::graph(const std::string& name) const
{
return SPODE::graph(name);

1
bayesnet/classifiers/SPODELd.h
View File

@@ -19,6 +19,7 @@ namespace bayesnet {
SPODELd& commonFit(const std::vector<std::string>& features, const std::string& className, map<std::string, std::vector<int>>& states, const Smoothing_t smoothing);
std::vector<std::string> graph(const std::string& name = "SPODELd") const override;
torch::Tensor predict(torch::Tensor& X) override;
torch::Tensor predict_proba(torch::Tensor& X) override;
static inline std::string version() { return "0.0.1"; };
};
}

5
bayesnet/classifiers/TANLd.cc
View File

@@ -29,6 +29,11 @@ namespace bayesnet {
auto Xt = prepareX(X);
return TAN::predict(Xt);
}
torch::Tensor TANLd::predict_proba(torch::Tensor& X)
{
auto Xt = prepareX(X);
return TAN::predict_proba(Xt);
}
std::vector<std::string> TANLd::graph(const std::string& name) const
{
return TAN::graph(name);

1
bayesnet/classifiers/TANLd.h
View File

@@ -18,6 +18,7 @@ namespace bayesnet {
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, const Smoothing_t smoothing) override;
std::vector<std::string> graph(const std::string& name = "TANLd") const override;
torch::Tensor predict(torch::Tensor& X) override;
torch::Tensor predict_proba(torch::Tensor& X) override;
};
}
#endif // !TANLD_H