Complete and fix KDB

2023-07-13 16:59:06 +02:00
parent 3fcf1e40c9
commit e52fdc718f
7 changed files with 75 additions and 37 deletions
--- a/src/Network.h
+++ b/src/Network.h
@@ -44,6 +44,7 @@ namespace bayesnet {
        torch::Tensor conditionalEdgeWeight();
        vector<vector<double>> predict_proba(const vector<vector<int>>&);
        double score(const vector<vector<int>>&, const vector<int>&);
+        vector<string> show();
        inline string version() { return "0.1.0"; }
    };
 }