Add predict_proba to Ld classifiers

2025-05-12 19:47:04 +02:00
parent 8a02a3a5cb
commit b11620bbe8
12 changed files with 116 additions and 44 deletions
--- a/bayesnet/classifiers/TANLd.cc
+++ b/bayesnet/classifiers/TANLd.cc
@@ -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);