From 24666a3a162cadeea2404e61fed849e05f9812d4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?= Date: Wed, 17 Jan 2024 10:47:20 +0100 Subject: [PATCH] Add number to b_list --- src/Platform/b_list.cc | 8 +++++--- src/Platform/modules/Models.h | 1 + src/Platform/modules/modelRegister.h | 2 ++ 3 files changed, 8 insertions(+), 3 deletions(-) diff --git a/src/Platform/b_list.cc b/src/Platform/b_list.cc index 00907aa..52b06d4 100644 --- a/src/Platform/b_list.cc +++ b/src/Platform/b_list.cc @@ -30,13 +30,15 @@ int main(int argc, char** argv) locale mylocale(std::cout.getloc(), new separated); locale::global(mylocale); std::cout.imbue(mylocale); - std::cout << Colors::GREEN() << "Dataset Sampl. Feat. Cls. Balance" << std::endl; + std::cout << Colors::GREEN() << " # Dataset Sampl. Feat. Cls. Balance" << std::endl; std::string balanceBars = std::string(BALANCE_LENGTH, '='); - std::cout << "============================== ====== ===== === " << balanceBars << std::endl; + std::cout << "=== ============================== ====== ===== === " << balanceBars << std::endl; bool odd = true; + int num = 0; for (const auto& dataset : data.getNames()) { auto color = odd ? Colors::CYAN() : Colors::BLUE(); - std::cout << color << setw(30) << left << dataset << " "; + std::cout << color << setw(3) << right << num++ << " "; + std::cout << setw(30) << left << dataset << " "; data.loadDataset(dataset); auto nSamples = data.getNSamples(dataset); std::cout << setw(6) << right << nSamples << " "; diff --git a/src/Platform/modules/Models.h b/src/Platform/modules/Models.h index a985c32..7c7c2a6 100644 --- a/src/Platform/modules/Models.h +++ b/src/Platform/modules/Models.h @@ -14,6 +14,7 @@ #include "STree.h" #include "ODTE.h" #include "SVC.h" +#include "XGBoost.h" #include "RandomForest.h" namespace platform { class Models { diff --git a/src/Platform/modules/modelRegister.h b/src/Platform/modules/modelRegister.h index 7f58401..bf69dd6 100644 --- a/src/Platform/modules/modelRegister.h +++ b/src/Platform/modules/modelRegister.h @@ -26,4 +26,6 @@ static platform::Registrar registrarSvc("SVC", [](void) -> bayesnet::BaseClassifier* { return new pywrap::SVC();}); static platform::Registrar registrarRaF("RandomForest", [](void) -> bayesnet::BaseClassifier* { return new pywrap::RandomForest();}); +static platform::Registrar registrarXGB("XGBoost", + [](void) -> bayesnet::BaseClassifier* { return new pywrap::XGBoost();}); #endif \ No newline at end of file