Add dump_cpt to Ensemble

This commit is contained in:
2025-02-12 20:55:35 +01:00
parent fb957ac3fe
commit f658149977
2 changed files with 9 additions and 4 deletions

View File

@@ -33,7 +33,12 @@ namespace bayesnet {
}
std::string dump_cpt() const override
{
return "";
std::string output;
for (auto& model : models) {
output += model->dump_cpt();
output += std::string(80, '-') + "\n";
}
return output;
}
protected:
torch::Tensor predict_average_voting(torch::Tensor& X);