Complete SPODE & AODE

2023-07-15 01:59:30 +02:00
parent db6908acd0
commit e311c27d43
7 changed files with 94 additions and 31 deletions
--- a/src/BaseClassifier.h
+++ b/src/BaseClassifier.h
@@ -31,6 +31,7 @@ namespace bayesnet {
        BaseClassifier& fit(vector<vector<int>>& X, vector<int>& y, vector<string>& features, string className, map<string, vector<int>>& states);
        void addNodes();
        Tensor predict(Tensor& X);
+        vector<int> predict(vector<vector<int>>& X);
        float score(Tensor& X, Tensor& y);
        float score(vector<vector<int>>& X, vector<int>& y);
        vector<string> show();