rmontanana/BayesNet
rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases 3 Wiki Activity

Fix mistake in node count

Browse Source
This commit is contained in:
Ricardo Montañana Gómez 2023-10-23 22:46:10 +02:00
parent 32293af69f
commit 4975feabff
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/BayesNet/Classifier.cc
View File

@ -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
{

4
src/Platform/ReportConsole.cc
View File

@ -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<string>().size());