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/Ensemble.h
+++ b/src/Ensemble.h
@@ -36,6 +36,7 @@ namespace bayesnet {
        float score(Tensor& X, Tensor& y);
        float score(vector<vector<int>>& X, vector<int>& y);
        vector<string> show();
+        vector<string> graph(string title);
    };
 }
 #endif