Add graphviz output to models

This commit is contained in:
2023-07-16 01:20:47 +02:00
parent 29aca0b35f
commit f530e69dae
16 changed files with 99 additions and 25 deletions

View File

@@ -93,11 +93,20 @@ namespace bayesnet {
}
vector<string> Ensemble::show()
{
vector<string> result;
auto result = vector<string>();
for (auto i = 0; i < n_models; ++i) {
auto res = models[i]->show();
result.insert(result.end(), res.begin(), res.end());
}
return result;
}
vector<string> Ensemble::graph(string title)
{
auto result = vector<string>();
for (auto i = 0; i < n_models; ++i) {
auto res = models[i]->graph(title + "_" + to_string(i));
result.insert(result.end(), res.begin(), res.end());
}
return result;
}
}