diff --git a/src/reports/ReportConsole.cpp b/src/reports/ReportConsole.cpp index 444a78c..a209c6d 100644 --- a/src/reports/ReportConsole.cpp +++ b/src/reports/ReportConsole.cpp @@ -140,12 +140,35 @@ namespace platform { } } } + line.str(""); + if (lastResult.find("score_train") == lastResult.end()) { + line << headerLine("Train score: -"); + } else { + line << headerLine("Train score: " + std::to_string(lastResult["score_train"].get())); + } + vbody.push_back(line.str()); sbody << line.str(); line.str(""); line << headerLine(fVector("Train scores: ", lastResult["scores_train"], 14, 12)); vbody.push_back(line.str()); sbody << line.str(); + line.str(""); line << headerLine("Test score: " + std::to_string(lastResult["score"].get())); + vbody.push_back(line.str()); sbody << line.str(); line.str(""); line << headerLine(fVector("Test scores: ", lastResult["scores_test"], 14, 12)); vbody.push_back(line.str()); sbody << line.str(); + line.str(""); + if (lastResult.find("train_time") == lastResult.end()) { + line << headerLine("Train time: -"); + } else { + line << headerLine("Train time: " + std::to_string(lastResult["train_time"].get())); + } + vbody.push_back(line.str()); sbody << line.str(); line.str(""); line << headerLine(fVector("Train times: ", lastResult["times_train"], 10, 3)); vbody.push_back(line.str()); sbody << line.str(); + line.str(""); + if (lastResult.find("test_time") == lastResult.end()) { + line << headerLine("Test time: -"); + } else { + line << headerLine("Test time: " + std::to_string(lastResult["test_time"].get())); + } + vbody.push_back(line.str()); sbody << line.str(); line.str(""); line << headerLine(fVector("Test times: ", lastResult["times_test"], 10, 3)); vbody.push_back(line.str()); sbody << line.str();