rmontanana/Platform
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add number to b_list

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-01-17 10:47:20 +01:00
parent 210ce4a255
commit 24666a3a16
3 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/Platform/b_list.cc
View File

@@ -30,13 +30,15 @@ int main(int argc, char** argv)
locale mylocale(std::cout.getloc(), new separated); locale mylocale(std::cout.getloc(), new separated);
locale::global(mylocale); locale::global(mylocale);
std::cout.imbue(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::string balanceBars = std::string(BALANCE_LENGTH, '=');
std::cout << "============================== ====== ===== === " << balanceBars << std::endl; std::cout << "=== ============================== ====== ===== === " << balanceBars << std::endl;
bool odd = true; bool odd = true;
int num = 0;
for (const auto& dataset : data.getNames()) { for (const auto& dataset : data.getNames()) {
auto color = odd ? Colors::CYAN() : Colors::BLUE(); 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); data.loadDataset(dataset);
auto nSamples = data.getNSamples(dataset); auto nSamples = data.getNSamples(dataset);
std::cout << setw(6) << right << nSamples << " "; std::cout << setw(6) << right << nSamples << " ";

1
src/Platform/modules/Models.h
View File

@@ -14,6 +14,7 @@
#include "STree.h" #include "STree.h"
#include "ODTE.h" #include "ODTE.h"
#include "SVC.h" #include "SVC.h"
#include "XGBoost.h"
#include "RandomForest.h" #include "RandomForest.h"
namespace platform { namespace platform {
class Models { class Models {

2
src/Platform/modules/modelRegister.h
View File

@@ -26,4 +26,6 @@ static platform::Registrar registrarSvc("SVC",
[](void) -> bayesnet::BaseClassifier* { return new pywrap::SVC();}); [](void) -> bayesnet::BaseClassifier* { return new pywrap::SVC();});
static platform::Registrar registrarRaF("RandomForest", static platform::Registrar registrarRaF("RandomForest",
[](void) -> bayesnet::BaseClassifier* { return new pywrap::RandomForest();}); [](void) -> bayesnet::BaseClassifier* { return new pywrap::RandomForest();});
static platform::Registrar registrarXGB("XGBoost",
[](void) -> bayesnet::BaseClassifier* { return new pywrap::XGBoost();});
#endif #endif