Add number to b_list

This commit is contained in:
2024-01-17 10:47:20 +01:00
parent 210ce4a255
commit 24666a3a16
3 changed files with 8 additions and 3 deletions

View File

@@ -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 << " ";

View File

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

View File

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