Add graphviz output to models

2023-07-16 01:20:47 +02:00
parent 29aca0b35f
commit f530e69dae
16 changed files with 99 additions and 25 deletions
--- a/src/BaseClassifier.h
+++ b/src/BaseClassifier.h
@@ -35,6 +35,7 @@ namespace bayesnet {
        float score(Tensor& X, Tensor& y);
        float score(vector<vector<int>>& X, vector<int>& y);
        vector<string> show();
+        virtual vector<string> graph(string title) = 0;
    };
 }
 #endif