diff --git a/src/BayesNet/Classifier.cc b/src/BayesNet/Classifier.cc index 7727939..a5b0420 100644 --- a/src/BayesNet/Classifier.cc +++ b/src/BayesNet/Classifier.cc @@ -137,7 +137,7 @@ namespace bayesnet { int Classifier::getNumberOfNodes() const { // Features does not include class - return fitted ? model.getFeatures().size() + 1 : 0; + return fitted ? model.getFeatures().size() : 0; } int Classifier::getNumberOfEdges() const { diff --git a/src/Platform/ReportConsole.cc b/src/Platform/ReportConsole.cc index c336d50..2479d06 100644 --- a/src/Platform/ReportConsole.cc +++ b/src/Platform/ReportConsole.cc @@ -29,8 +29,8 @@ namespace platform { void ReportConsole::body() { auto tmp = ConfigLocale(); - int maxHyper = 0; - int maxDataset = 0; + int maxHyper = 15; + int maxDataset = 7; for (const auto& r : data["results"]) { maxHyper = max(maxHyper, (int)r["hyperparameters"].dump().size()); maxDataset = max(maxDataset, (int)r["dataset"].get().size());